Update app.py

app.py

@@ -36,46 +36,56 @@ def parse_conversation(messages):
     return system_prompt, initial_query
 
 def respond(message, history, model, approach, system_message, max_tokens, temperature, top_p):
-    client = OpenAI(api_key=API_KEY, base_url="https://openrouter.ai/api/v1")
-    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]})
-    messages.append({"role": "user", "content": message})
-
-    if approach == "none":
-        response = client.chat.completions.create(
-            extra_headers={
-            "HTTP-Referer": "https://github.com/codelion/optillm",
-            "X-Title": "optillm"    
-            },
-            model=model,
-            messages=messages,
-            max_tokens=max_tokens,
-            temperature=temperature,
-            top_p=top_p,
-        )
-        return response.choices[0].message.content
-    else:
-        system_prompt, initial_query = parse_conversation(messages)
-        if approach == 'rto':
-            final_response, _ = round_trip_optimization(system_prompt, initial_query, client, model)
-        elif approach == 'z3':
-            z3_solver = Z3SymPySolverSystem(system_prompt, client, model)
-            final_response, _ = z3_solver.process_query(initial_query)
-        elif approach == "self_consistency":
-            final_response, _ = advanced_self_consistency_approach(system_prompt, initial_query, client, model)
-        elif approach == "cot_reflection":
-            final_response, _ = cot_reflection(system_prompt, initial_query, client, model)
-        elif approach == 'plansearch':
-            response, _ = plansearch(system_prompt, initial_query, client, model)
-            final_response = response[0]
-        elif approach == 'leap':
-            final_response, _ = leap(system_prompt, initial_query, client, model)
-        elif approach == 're2':
-            final_response, _ = re2_approach(system_prompt, initial_query, client, model)
+    try:
+        client = OpenAI(api_key=API_KEY, base_url="https://openrouter.ai/api/v1")
+        messages = [{"role": "system", "content": system_message}]
         
-    return final_response
+        for val in history:
+            if val[0]:
+                messages.append({"role": "user", "content": val[0]})
+            if val[1]:
+                messages.append({"role": "assistant", "content": val[1]})
+        
+        messages.append({"role": "user", "content": message})
+        
+        if approach == "none":
+            response = client.chat.completions.create(
+                extra_headers={
+                    "HTTP-Referer": "https://github.com/codelion/optillm",
+                    "X-Title": "optillm"
+                },
+                model=model,
+                messages=messages,
+                max_tokens=max_tokens,
+                temperature=temperature,
+                top_p=top_p,
+            )
+            return response.choices[0].message.content
+        else:
+            system_prompt, initial_query = parse_conversation(messages)
+            
+            if approach == 'rto':
+                final_response, _ = round_trip_optimization(system_prompt, initial_query, client, model)
+            elif approach == 'z3':
+                z3_solver = Z3SymPySolverSystem(system_prompt, client, model)
+                final_response, _ = z3_solver.process_query(initial_query)
+            elif approach == "self_consistency":
+                final_response, _ = advanced_self_consistency_approach(system_prompt, initial_query, client, model)
+            elif approach == "cot_reflection":
+                final_response, _ = cot_reflection(system_prompt, initial_query, client, model)
+            elif approach == 'plansearch':
+                response, _ = plansearch(system_prompt, initial_query, client, model)
+                final_response = response[0]
+            elif approach == 'leap':
+                final_response, _ = leap(system_prompt, initial_query, client, model)
+            elif approach == 're2':
+                final_response, _ = re2_approach(system_prompt, initial_query, client, model)
+            
+            return final_response
+            
+    except Exception as e:
+        error_message = f"Error in respond function: {str(e)}\nType: {type(e).__name__}"
+        print(error_message)
 
     # for message in client.chat_completion(
     #     messages,