Update app.py

app.py

@@ -63,44 +63,79 @@ import gradio as gr
 from dotenv import load_dotenv
 from huggingface_hub import InferenceClient
 
+# Load environment variables
 load_dotenv()
 
 HF_TOKEN = os.getenv("HF_TOKEN")
+
 if not HF_TOKEN:
     raise ValueError("HF_TOKEN is missing.")
 
+# Create HF client
+client = InferenceClient(
+    provider="nscale",
+    api_key=HF_TOKEN
+)
+
+# System prompt
 system_message = (
     "You are a helpful and experienced coding assistant specialized in web development. "
     "Help the user by generating complete and functional code for building websites. "
-    "You can provide HTML, CSS, JavaScript, and backend code (like Flask, Node.js, etc.) "
+    "You can provide HTML, CSS, JavaScript, and backend code like Flask, Node.js, etc. "
     "based on their requirements."
 )
 
-client = InferenceClient(
-    provider="nscale",
-    api_key=HF_TOKEN,
-)
-
+# Chat function
 def chat_function(message, history):
-    messages = [{"role": "system", "content": system_message}]
+
+    messages = [
+        {
+            "role": "system",
+            "content": system_message
+        }
+    ]
 
     history = history or []
+
     for item in history:
+
+        # Modern Gradio messages format
         if isinstance(item, dict):
+
             role = item.get("role")
             content = item.get("content", "")
-            if role in ("user", "assistant"):
-                messages.append({"role": role, "content": content})
+
+            if role in ["user", "assistant"]:
+                messages.append({
+                    "role": role,
+                    "content": content
+                })
+
+        # Older tuple fallback
         elif isinstance(item, (list, tuple)) and len(item) == 2:
+
             user_msg, assistant_msg = item
+
             if user_msg:
-                messages.append({"role": "user", "content": user_msg})
+                messages.append({
+                    "role": "user",
+                    "content": user_msg
+                })
+
             if assistant_msg:
-                messages.append({"role": "assistant", "content": assistant_msg})
+                messages.append({
+                    "role": "assistant",
+                    "content": assistant_msg
+                })
 
-    messages.append({"role": "user", "content": message})
+    # Current user message
+    messages.append({
+        "role": "user",
+        "content": message
+    })
 
     try:
+
         completion = client.chat.completions.create(
             model="Qwen/Qwen2.5-Coder-7B-Instruct",
             messages=messages,
@@ -108,16 +143,20 @@ def chat_function(message, history):
             temperature=0.7,
             top_p=0.95,
         )
+
         return completion.choices[0].message.content
+
     except Exception as e:
         return f"Error: {str(e)}"
 
+# Interface
 demo = gr.ChatInterface(
     fn=chat_function,
     type="messages",
     title="AI Coding Assistant",
-    description="A coding assistant powered by Qwen2.5-Coder.",
+    description="A coding assistant powered by Qwen2.5-Coder."
 )
 
+# Launch
 if __name__ == "__main__":
     demo.launch()