arreglos 4

app.py

@@ -7,33 +7,48 @@ client = InferenceClient("AuriLab/gpt-bi-instruct-cesar")
 def format_messages(history: List[Tuple[str, str]], system_message: str, user_message: str) -> List[Dict[str, str]]:
     messages = [{"role": "system", "content": system_message}]
     messages.extend([
-        {"role": "user" if i % 2 == 0 else "assistant", "content": msg}
+        {"role": "user" if i % 2 == 0 else "assistant", "content": str(msg)}  # Convert msg to string
         for turn in history
         for i, msg in enumerate(turn)
-        if msg
+        if msg is not None
     ])
-    messages.append({"role": "user", "content": user_message})
+    messages.append({"role": "user", "content": str(user_message)})  # Convert user_message to string
     return messages
 
-def respond(message: str, history: List[Tuple[str, str]], system_message: str, max_tokens: int, temperature: float, top_p: float) -> str:
+def respond(message: str, history: List[Tuple[str, str]]) -> str:
+    # Default values for parameters
+    system_message = "You are a helpful AI assistant."
+    max_tokens = 1000
+    temperature = 0.7
+    top_p = 0.85
+    
     messages = format_messages(history, system_message, message)
     response = ""
     
-    for msg in client.chat_completion(
-        messages,
-        max_tokens=max_tokens,
-        stream=True,
-        temperature=0.7,        # Aumentado para más variedad
-        top_p=0.85,            # Ajustado para mejor balance
-    ):
-        token = msg.choices[0].delta.content
-        response += token
-        yield response
+    try:
+        for msg in client.chat_completion(
+            messages,
+            max_tokens=max_tokens,
+            stream=True,
+            temperature=temperature,
+            top_p=top_p,
+        ):
+            if hasattr(msg.choices[0].delta, 'content'):
+                token = msg.choices[0].delta.content
+                if token is not None:
+                    response += token
+                    yield response
+    except Exception as e:
+        return f"Error: {str(e)}"
 
 demo = gr.ChatInterface(
-    respond,
+    fn=respond,
     title="Demo GPT-BI instruct",
+    examples=["nola duzu izena?", "Nola egiten duzu?"],
+    retry_btn="Retry",
+    undo_btn="Undo",
+    clear_btn="Clear",
 )
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch(share=False)