Update app.py

app.py

@@ -8,72 +8,68 @@ DEFAULT_SYSTEM_PROMPT = (
     "ROLE:\n"
     "You are an Elite Minecraft Modding and Environment Diagnostician.\n"
     "The user is an experienced modder. Do NOT provide basic IT advice.\n\n"
-
     "GOAL:\n"
     "Analyze Minecraft crash reports (latest.log, crash-reports, hs_err_pid) and pinpoint the exact root cause.\n"
     "Identify whether the issue is caused by a specific mod, mod conflict, library, JVM configuration, or hardware/driver issue.\n\n"
-
     "KNOWLEDGE DOMAINS:\n"
     "1. Java/JVM & Environment:\n"
     "- Detect incorrect Java versions (Java 8 vs 17 vs 21 mismatches).\n"
     "- Identify GC overhead errors, invalid JVM flags, and memory leaks.\n"
     "- Diagnose hs_err_pid crashes (native crashes, lwjgl corruption).\n\n"
-
     "2. Hardware & OS:\n"
     "- Diagnose EXCEPTION_ACCESS_VIOLATION (0xc0000005).\n"
     "- Detect GL_OUT_OF_MEMORY and OpenGL issues.\n"
     "- Identify wrong GPU usage (integrated vs dedicated).\n"
     "- Detect outdated or incompatible graphics drivers.\n\n"
-
     "3. Modloader & Mixins:\n"
     "- Trace NoSuchMethodError, ClassNotFoundException, AbstractMethodError to exact modid.\n"
     "- Analyze Mixin conflicts, failed injections, OVERWRITE/REDIRECT failures.\n"
     "- Detect duplicate or incompatible mods.\n\n"
-
     "4. Rendering Conflicts:\n"
     "- Detect conflicts between OptiFine, Sodium/Rubidium/Embeddium, Iris/Oculus.\n\n"
-
     "5. Data/Config Issues:\n"
     "- Detect malformed JSON configs.\n"
     "- Identify corrupted world data (level.dat).\n"
     "- Detect missing dependencies or broken datapacks.\n\n"
-
     "CONSTRAINTS:\n"
     "- Do NOT guess.\n"
     "- Do NOT hallucinate mod names or file paths.\n"
     "- If stack trace shows net.minecraft.*, scan deeper to find injected mod.\n"
     "- In Mixin conflicts, identify the actual instigator.\n"
     "- Be concise and technical.\n\n"
-
     "OUTPUT FORMAT:\n"
     "1. Root Cause: One-line exact failure source.\n"
     "2. Evidence: Exact log line, stack trace, or error code.\n"
     "3. Fix: Direct actionable solution.\n"
 )
 
-def respond(message, history, system_message, max_tokens, temperature, top_p, hf_token: gr.OAuthToken):
-    client = InferenceClient(token=hf_token.token if hf_token else None, model=MODEL_ID)
+def respond(message, history, system_message, max_tokens, temperature, top_p):
+    # This retrieves the token from the session automatically if logged in via gr.LoginButton
+    client = InferenceClient(model=MODEL_ID)
     
     messages = [{"role": "system", "content": system_message}]
-    for val in history:
-        if val[0]: messages.append({"role": "user", "content": val[0]})
-        if val[1]: messages.append({"role": "assistant", "content": val[1]})
+    for user_msg, assistant_msg in history:
+        if user_msg: messages.append({"role": "user", "content": user_msg})
+        if assistant_msg: messages.append({"role": "assistant", "content": assistant_msg})
     messages.append({"role": "user", "content": message})
 
     response = ""
     try:
-        # Qwen models are very fast on the HF API
         for message_chunk in client.chat_completion(
-            messages, max_tokens=max_tokens, stream=True, temperature=temperature, top_p=top_p
+            messages, 
+            max_tokens=max_tokens, 
+            stream=True, 
+            temperature=temperature, 
+            top_p=top_p
         ):
             token = message_chunk.choices[0].delta.content or ""
             response += token
             yield response
     except Exception as e:
-        yield f"API Error: {str(e)}. Check if your HF Token is valid or if the model is currently loading."
+        yield f"API Error: {str(e)}. Make sure you are logged in via the Sidebar."
 
 # Gradio Interface
-with gr.Blocks(theme=gr.themes.Monochrome()) as demo:
+with gr.Blocks() as demo:
     gr.Markdown("# 🛠️ Minecraft Modding Log Analyzer")
     
     with gr.Sidebar():
@@ -81,13 +77,16 @@ with gr.Blocks(theme=gr.themes.Monochrome()) as demo:
         gr.Markdown("---")
         system_msg = gr.Textbox(value=DEFAULT_SYSTEM_PROMPT, label="Modder Persona", lines=5)
         tokens = gr.Slider(128, 8192, value=2048, label="Max Log Length")
-        temp = gr.Slider(0.1, 1.0, value=0.3, label="Creativity (Keep low for logs)")
+        temp = gr.Slider(0.1, 1.0, value=0.3, label="Creativity")
         top_p = gr.Slider(0.1, 1.0, value=0.9, label="Top-P")
 
+    # The additional_inputs must exactly match the arguments in respond() 
+    # after 'message' and 'history'
     gr.ChatInterface(
         respond,
         additional_inputs=[system_msg, tokens, temp, top_p],
     )
 
 if __name__ == "__main__":
-    demo.launch()
+    # Theme is applied here for Gradio 6.0 compatibility
+    demo.launch(theme=gr.themes.Monochrome())