Update app.py

app.py

@@ -3,25 +3,28 @@ import gradio as gr
 import json
 from huggingface_hub import InferenceClient
 
-# Load environment token
+# Load your Hugging Face token from environment variable
 HF_TOKEN = os.getenv("HF_API_TOKEN")
+
+# Initialize the inference client
 client = InferenceClient(
     model="deepseek-ai/DeepSeek-V3",
     token=HF_TOKEN
 )
 
-# Load word and language data
+# Load Quran words and language list
 with open("top_500_quran_lemmas_fixed.json", encoding="utf-8") as f:
     word_list = json.load(f)
 
 with open("language_list.json", encoding="utf-8") as f:
     language_list = json.load(f)
 
+# Prepare dropdown options
 word_options = [f"{word['text']} ({word['english']})" for word in word_list]
 language_options = [f"{lang['name']} ({lang['code']})" for lang in language_list]
 
 
-def create_messages(word_entry, language_code):
+def create_messages(word_entry, language_name):
     return [
         {
             "role": "system",
@@ -30,13 +33,13 @@ def create_messages(word_entry, language_code):
         {
             "role": "user",
             "content": f"""
-Explain the Quranic word "{word_entry['text']}" (which means "{word_entry['english']}") in {language_code}.
+Explain the Quranic word "{word_entry['text']}" (which means "{word_entry['english']}") in {language_name}.
 
 Please include:
-1. Translation in {language_code}
+1. Translation in {language_name}
 2. Root word and derivatives
 3. Occurrences in the Qur'an (Surah & Verse)
-4. Explanation of each occurrence using easy language
+4. Explanation of each occurrence using easy-to-understand {language_name}
 """,
         },
     ]
@@ -44,36 +47,44 @@ Please include:
 
 def process(word_label, lang_label):
     selected_word = next((w for w in word_list if w['text'] in word_label), None)
-    language_code = lang_label.split("(")[-1].strip(")")
+    language_name = lang_label.split("(")[1].strip()
 
     if not selected_word:
-        return "❌ Word not found."
+        yield "❌ Word not found."
+        return
 
-    messages = create_messages(selected_word, language_code)
+    messages = create_messages(selected_word, language_name)
 
     try:
-        response = client.chat.completions.create(
+        stream = client.chat.completions.create(
             messages=messages,
-            temperature=0.7,
-            top_p=0.9,
-            max_tokens=1024,
-            stream=False
+            temperature=0.3,
+            top_p=0.7,
+            max_tokens=128,
+            stream=True
         )
-        return response.choices[0].message.content
+
+        output = ""
+        for chunk in stream:
+            if chunk.choices and chunk.choices[0].delta.content:
+                output += chunk.choices[0].delta.content
+                yield output
 
     except Exception as e:
-        return f"❌ Error: {e}"
+        yield f"❌ Error: {e}"
 
 
-# Gradio UI
+# Build Gradio UI
 with gr.Blocks() as demo:
-    gr.Markdown("## Quran Word Info (Powered by DeepSeek-V3)")
+    gr.Markdown("## 📖 Quran Word Explorer (with DeepSeek-V3) — Streaming Enabled")
+
     with gr.Row():
-        word_input = gr.Dropdown(choices=word_options, label="Select Word")
-        lang_input = gr.Dropdown(choices=language_options, label="Select Language")
-    run_btn = gr.Button("Get Info")
-    output = gr.Textbox(lines=20, label="Response")
+        word_input = gr.Dropdown(choices=word_options, label="🔤 Select Quranic Word")
+        lang_input = gr.Dropdown(choices=language_options, label="🌐 Select Language")
+
+    run_btn = gr.Button("🔍 Get Explanation")
+    output = gr.Textbox(label="📘 Output", lines=20)
 
-    run_btn.click(fn=process, inputs=[word_input, lang_input], outputs=output)
+    run_btn.click(fn=process, inputs=[word_input, lang_input], outputs=output, streaming=True)
 
-demo.launch()
+demo.launch()