Update app.py

app.py

@@ -12,20 +12,24 @@ def resolve_token(ui_token):
     if ui_token and ui_token.strip():
         return ui_token.strip()
 
-    env_token = os.getenv("hf")
+    env_token = os.getenv("HUGGINGFACEHUB_API_TOKEN")
     if env_token:
         return env_token.strip()
 
+    return None
+
+
 SUPPORTED_EXT = (
     ".pdf", ".docx", ".txt", ".png", ".jpg", ".jpeg", ".webp", ".bmp", ".tiff"
 )
 
+
 def extract_text_from_file(filepath):
     if not filepath:
         return ""
 
-    if hasattr(filepath, "name"):
-        filepath = filepath.name
+    if isinstance(filepath, dict) and "name" in filepath:
+        filepath = filepath["name"]
 
     ext = pathlib.Path(filepath).suffix.lower()
 
@@ -52,20 +56,18 @@ def extract_text_from_file(filepath):
     except Exception as e:
         return f"Error reading file: {str(e)}"
 
+
 MODELS = {
     "Qwen 3.5 0.8B (Fastest)": "Qwen/Qwen3.5-0.8B",
     "DeepSeek R1 1.5B": "deepseek-ai/DeepSeek-R1-Distill-Qwen-1.5B",
     "Qwen 3.5 2B (Balanced Speed)": "Qwen/Qwen3.5-2B",
-
     "Gemma 4 5B": "google/gemma-4-E2B-it",
     "Qwen 3.5 4B": "Qwen/Qwen3.5-4B",
     "Gemma 4 9B": "google/gemma-4-9b-it",
-
     "DeepSeek Qwen 7B": "deepseek-ai/DeepSeek-R1-Distill-Qwen-7B",
     "DeepSeek R1 Llama 8B": "deepseek-ai/DeepSeek-R1-Distill-Llama-8B",
     "Gemma 8B": "google/gemma-4-E4B-it",
     "Qwen 3.5 9B (Balanced Thinking)": "Qwen/Qwen3.5-9B",
-
     "Qwen 3.6 27B (Best)": "Qwen/Qwen3.6-27B",
 }
 
@@ -76,15 +78,14 @@ Follow instructions strictly.
 Use simple language.
 Be structured.
 Avoid repetition.
-Output clean markdown.
+Output markdown.
 """
 
-def make_prompts(topic):
 
+def make_prompts(topic):
     base = f"""
 Topic:
 {topic}
-
 Instructions:
 - Be clear and concise
 - Use bullet points
@@ -92,10 +93,8 @@ Instructions:
 """
 
     return {
-
         "lesson": base + """
 Create a lesson plan with:
-
 1. Objectives (3-5 points)
 2. Short introduction
 3. Key concepts (bullet points)
@@ -104,48 +103,47 @@ Create a lesson plan with:
 6. 1 classroom activity
 7. 5 assessment questions
 """,
-
         "qa": base + """
 Generate 10 short exam questions with answers.
 Keep answers brief (2-3 lines each).
 """,
-
         "mcq": base + """
 Generate 10 MCQs:
 - 4 options each
 - mark correct answer
 """,
-
         "summary": base + """
 Write a 200-word summary.
 Use simple sentences.
 """
     }
 
-def stream_llm(model_id, prompt, hf_token):
 
+def stream_llm(model_id, prompt, hf_token):
     if not hf_token:
         yield "❌ No Hugging Face API key found."
         return
 
     try:
-        client = InferenceClient(
-            model=model_id,
-            token=hf_token,
-        )
+        client = InferenceClient(model=model_id, token=hf_token)
 
         stream = client.text_generation(
             SYSTEM_MSG + "\n\n" + prompt,
             max_new_tokens=1024,
-            temperature=0.5,
-            top_p=0.9,
+            temperature=0.7,
+            top_p=0.95,
             repetition_penalty=1.1,
             stream=True,
         )
 
         partial = ""
-        for token in stream:
-            partial += token
+        for chunk in stream:
+            if hasattr(chunk, "token"):
+                token_text = chunk.token.text
+            else:
+                token_text = str(chunk)
+
+            partial += token_text
             yield partial
 
     except Exception as e:
@@ -155,8 +153,8 @@ def stream_llm(model_id, prompt, hf_token):
         else:
             yield f"❌ API Error:\n{err}"
 
-def generate_content(text, file, model_label, token):
 
+def generate_content(text, file, model_label, token):
     file_text = extract_text_from_file(file) if file else ""
     syllabus = (text + "\n\n" + file_text).strip()
 
@@ -175,6 +173,7 @@ def generate_content(text, file, model_label, token):
             outputs[i] = chunk
             yield tuple(outputs)
 
+
 CSS = """
 body,.gradio-container{
     font-family: Inter, sans-serif !important;
@@ -185,7 +184,6 @@ with gr.Blocks(css=CSS) as demo:
     gr.Markdown("# 🎓 AI Study Material Generator (Streaming)")
 
     with gr.Row():
-
         with gr.Column():
             text_input = gr.Textbox(
                 placeholder="Paste syllabus or topic",
@@ -208,7 +206,6 @@ with gr.Blocks(css=CSS) as demo:
             btn = gr.Button("Generate")
 
     with gr.Tabs():
-
         with gr.TabItem("Lesson Plan"):
             lesson = gr.Markdown()