Update app.py

app.py

@@ -2,12 +2,6 @@ import gradio as gr
 import torch
 from transformers import pipeline, AutoTokenizer, AutoModelForSeq2SeqLM
 import re
-import os
-from huggingface_hub import login
-
-# Authenticate with Hugging Face
-if "HF_TOKEN" in os.environ:
-    login(token=os.environ["HF_TOKEN"])
 
 # Global variables to store the models
 atlas_pipe = None
@@ -225,17 +219,27 @@ def chat_with_atlas(message, history):
         # Detect if input is Arabizi
         is_arabizi_input = detect_arabizi(message)
         
+        print("\n" + "="*60)
+        print("🔍 DEBUG LOG - FULL CONVERSION PIPELINE")
+        print("="*60)
+        print(f"📥 ORIGINAL INPUT: '{message}'")
+        print(f"🤖 ARABIZI DETECTED: {is_arabizi_input}")
+        
         # Prepare input for the model
         if is_arabizi_input:
             # Convert Arabizi to Arabic using AI model
-            print(f"🔄 Converting Arabizi: '{message}'")
+            print(f"\n🔄 STEP 1: Converting Arabizi to Arabic...")
             arabic_input = arabizi_to_arabic_ai(message)
-            print(f"✅ Converted to Arabic: '{arabic_input}'")
+            print(f"✅ ARABIC CONVERSION: '{arabic_input}'")
             model_input = arabic_input
         else:
             # Use original input (Arabic or English)
+            print(f"\n➡️  NO CONVERSION NEEDED - Using original input")
             model_input = message
         
+        print(f"\n🤖 STEP 2: Sending to Atlas-Chat model...")
+        print(f"📤 MODEL INPUT: '{model_input}'")
+        
         # Generate response using Arabic input
         messages = [{"role": "user", "content": model_input}]
         
@@ -249,19 +253,28 @@ def chat_with_atlas(message, history):
         
         # Extract the response
         response = outputs[0]["generated_text"][-1]["content"].strip()
-        print(f"🤖 Atlas response: '{response}'")
+        print(f"✅ MODEL RESPONSE (Arabic): '{response}'")
         
         # Convert response back to Arabizi if input was Arabizi
         if is_arabizi_input:
+            print(f"\n🔄 STEP 3: Converting response back to Arabizi...")
             arabizi_response = arabic_to_arabizi(response)
-            print(f"🔄 Converted to Arabizi: '{arabizi_response}'")
+            print(f"✅ FINAL ARABIZI RESPONSE: '{arabizi_response}'")
+            print("="*60)
+            print("🎯 FINAL OUTPUT TO USER:", arabizi_response)
+            print("="*60 + "\n")
             return arabizi_response
         else:
             # Return original response for Arabic/English
+            print(f"\n➡️  NO BACK-CONVERSION NEEDED")
+            print("="*60)
+            print("🎯 FINAL OUTPUT TO USER:", response)
+            print("="*60 + "\n")
             return response
         
     except Exception as e:
-        print(f"❌ Error in chat: {e}")
+        print(f"\n❌ ERROR OCCURRED: {str(e)}")
+        print("="*60 + "\n")
         # Return error in appropriate language
         if detect_arabizi(message):
             return f"sorry, kan chi mochkil: {str(e)}. 3awd jar'b!"