Update app.py

app.py

@@ -12,15 +12,23 @@ MODELS = {
     "BioTransformer 1B": "omurberaisik/biotransformer-1b-v1",
 }
 
+def parse_history(history):
+    messages = []
+    for item in history:
+        if isinstance(item, dict):
+            messages.append({"role": item["role"], "content": item["content"]})
+        else:
+            user_msg, bot_msg = item
+            messages.append({"role": "user", "content": user_msg})
+            if bot_msg:
+                messages.append({"role": "assistant", "content": bot_msg})
+    return messages
+
 def respond(message, history, model_name, max_tokens, temperature, top_p):
     model_id = MODELS[model_name]
     client = InferenceClient(model_id)
 
-    messages = []
-    for user_msg, bot_msg in history:
-        messages.append({"role": "user", "content": user_msg})
-        if bot_msg:
-            messages.append({"role": "assistant", "content": bot_msg})
+    messages = parse_history(history)
     messages.append({"role": "user", "content": message})
 
     response = ""
@@ -38,10 +46,9 @@ def respond(message, history, model_name, max_tokens, temperature, top_p):
                 yield response
     except Exception:
         prompt = ""
-        for user_msg, bot_msg in history:
-            prompt += f"User: {user_msg}\n"
-            if bot_msg:
-                prompt += f"Assistant: {bot_msg}\n"
+        for item in messages[:-1]:
+            role = "User" if item["role"] == "user" else "Assistant"
+            prompt += f"{role}: {item['content']}\n"
         prompt += f"User: {message}\nAssistant:"
         try:
             for chunk in client.text_generation(