Update app.py

app.py

@@ -22,7 +22,7 @@ class Config:
     HISTORY_FILE = "chat_history.csv"
     CONFIG_FILE = "config.json"
     DEFAULT_SETTINGS = {
-        "language": "ar",  # Default to Arabic
+        "language": "ar",
         "theme": THEME,
         "max_tokens": MAX_TOKENS,
         "temperature": TEMPERATURE,
@@ -47,9 +47,11 @@ class Config:
 def load_model():
     """Load the model and tokenizer with GPU support if available"""
     try:
-        tokenizer = AutoTokenizer.from_pretrained(Config.MODEL_ID)
+        tokenizer = AutoTokenizer.from_pretrained(
+            Config.MODEL_ID,
+            trust_remote_code=True
+        )
         
-        # Device configuration
         if torch.cuda.is_available():
             device = "cuda"
             torch_dtype = torch.float16
@@ -62,7 +64,8 @@ def load_model():
         model = AutoModelForCausalLM.from_pretrained(
             Config.MODEL_ID,
             torch_dtype=torch_dtype,
-            device_map="auto" if device == "cuda" else None
+            device_map="auto" if device == "cuda" else None,
+            trust_remote_code=True
         )
         
         if device == "cpu":
@@ -159,7 +162,7 @@ You are Edraky (إدراكي), a smart, multilingual AI assistant built to suppo
 
 🎓 Key Principles:
 
-1. Use simplified language that students of all levels can understand—don’t overcomplicate.
+1. Use simplified language that students of all levels can understand—don't overcomplicate.
 2. Always encourage curiosity, even if a question is simple or incorrect.
 3. Teach in both Arabic and English, depending on the user's input. If the user speaks Arabic, prefer using Modern Standard Arabic unless otherwise specified.
 4. Be culturally aware of Egypt and the Arab world. Use relevant examples when explaining.
@@ -198,12 +201,9 @@ You are Edraky (إدراكي), a smart, multilingual AI assistant built to suppo
             pad_token_id=tokenizer.eos_token_id
         )
 
-        # The output structure may depend on the pipeline/model; adjust as needed
-        # Here, we assume outputs[0]['generated_text'][-1]['content'] is correct
         try:
             response = outputs[0]['generated_text'][-1]['content']
         except Exception:
-            # Fallback for different output structure
             response = outputs[0]['generated_text'] if 'generated_text' in outputs[0] else str(outputs)
         log_conversation(prompt, response, language)
         return response
@@ -232,12 +232,10 @@ def analyze_history() -> Optional[plt.Figure]:
             
         plt.figure(figsize=(12, 8))
         
-        # Create subplots
         ax1 = plt.subplot2grid((2, 2), (0, 0))
         ax2 = plt.subplot2grid((2, 2), (0, 1))
         ax3 = plt.subplot2grid((2, 2), (1, 0), colspan=2)
         
-        # Plot 1: Daily conversation count
         df['date'] = df['timestamp'].dt.date
         daily_counts = df.groupby('date').size()
         daily_counts.plot(kind='bar', ax=ax1, color='#4e79a7')
@@ -245,7 +243,6 @@ def analyze_history() -> Optional[plt.Figure]:
         ax1.set_xlabel("Date")
         ax1.set_ylabel("Count")
         
-        # Plot 2: Language distribution
         lang_dist = df['language'].value_counts()
         lang_dist.plot(kind='pie', ax=ax2, autopct='%1.1f%%', 
                       colors=['#f28e2b', '#e15759'], 
@@ -253,7 +250,6 @@ def analyze_history() -> Optional[plt.Figure]:
         ax2.set_title("Language Distribution")
         ax2.set_ylabel("")
         
-        # Plot 3: Hourly activity
         df['hour'] = df['timestamp'].dt.hour
         hourly = df.groupby('hour').size()
         sns.lineplot(x=hourly.index, y=hourly.values, ax=ax3, color='#59a14f')
@@ -389,7 +385,6 @@ def create_chat_tab(settings: dict):
                 lines=10
             )
     
-    # Event handlers
     submit_btn.click(
         fn=lambda p, l, m, t, tp: generate_response(
             p, 
@@ -415,7 +410,6 @@ def create_history_tab():
         clear_btn = gr.Button("🗑️ مسح السجل / Clear History", variant="stop")
         export_btn = gr.Button("📤 تصدير / Export")
     
-    # Event handlers
     refresh_btn.click(fn=show_history, outputs=history_table)
     clear_btn.click(fn=clear_history, outputs=history_table)
     export_btn.click(
@@ -434,7 +428,6 @@ def create_analytics_tab():
     with gr.Row():
         refresh_analytics = gr.Button("🔄 تحديث التحليلات / Refresh Analytics")
     
-    # Event handlers
     refresh_analytics.click(fn=analyze_history, outputs=analytics_plot)
 
 def create_settings_tab(settings: dict):
@@ -476,7 +469,6 @@ def create_settings_tab(settings: dict):
         save_btn = gr.Button("💾 حفظ الإعدادات / Save Settings", variant="primary")
         status = gr.Textbox(label="الحالة / Status", interactive=False)
     
-    # Event handler
     save_btn.click(
         fn=lambda t, l, m, temp, tp: save_settings(t, l, m, temp, tp, status),
         inputs=[theme, default_lang, default_max_tokens, default_temp, default_top_p],
@@ -503,28 +495,21 @@ def create_interface():
     current_settings = Config.load_settings()
     
     with gr.Blocks(theme=current_settings["theme"], css=get_css()) as app:
-        # Header section
         create_header()
         
-        # Main tabs
         with gr.Tabs():
-            # Chat tab
             with gr.Tab("💬 Chat", id="chat"):
                 create_chat_tab(current_settings)
             
-            # History tab
             with gr.Tab("📜 History", id="history"):
                 create_history_tab()
             
-            # Analytics tab
             with gr.Tab("📊 Analytics", id="analytics"):
                 create_analytics_tab()
             
-            # Settings tab
             with gr.Tab("⚙️ Settings", id="settings"):
                 create_settings_tab(current_settings)
         
-        # Footer
         create_footer()
         
         return app
@@ -533,10 +518,8 @@ def create_interface():
 # MAIN EXECUTION
 # ======================
 if __name__ == "__main__":
-    # Ensure necessary directories exist
     Path("logs").mkdir(exist_ok=True)
     
-    # Create and launch interface
     app = create_interface()
     app.launch(
         server_name="0.0.0.0",