Update app.py

app.py

@@ -12,7 +12,7 @@ except Exception as e:
 
 def ai_vote(poll_title, choices, num_ais):
     if chat_pipeline is None:
-        return {"Error": "Model not initialized"}, []
+        return "Error: Model not initialized.", ""
 
     # Initialize results and explanations
     results = {choice: 0 for choice in choices}
@@ -23,7 +23,6 @@ def ai_vote(poll_title, choices, num_ais):
         input_text = f"Poll Title: {poll_title}\nChoices: {', '.join(choices)}\nChoose the best option and explain why."
         try:
             response = chat_pipeline(input_text, max_length=150, num_return_sequences=1)[0]['generated_text']
-            # Extract the choice and explanation
             for choice in choices:
                 if choice.lower() in response.lower():
                     results[choice] += 1
@@ -31,17 +30,30 @@ def ai_vote(poll_title, choices, num_ais):
                     explanations.append((choice, explanation))
                     break
         except Exception as e:
-            return {"Error": str(e)}, []
+            return f"Error: {str(e)}", ""
 
-    return results, explanations
+    # Convert results to HTML for styled output
+    styled_results = f"<h2>{poll_title}</h2>"
+    styled_results += "<ul>"
+    for choice, votes in results.items():
+        styled_results += f"<li><strong>{choice}</strong>: {votes} votes</li>"
+    styled_results += "</ul>"
+
+    # Add explanations in HTML
+    styled_results += "<h3>AI Explanations:</h3><ul>"
+    for choice, explanation in explanations:
+        styled_results += f"<li><strong>{choice}:</strong> {explanation}</li>"
+    styled_results += "</ul>"
+
+    return styled_results, explanations
 
 def gradio_interface(title, choices, num_ais):
     try:
         choices = [choice.strip() for choice in choices.split(",")]
-        results, explanations = ai_vote(title, choices, num_ais)
-        return results, explanations
+        styled_results, explanations = ai_vote(title, choices, num_ais)
+        return styled_results, explanations
     except Exception as e:
-        return {"Error": str(e)}, "An error occurred."
+        return f"Error: {str(e)}", ""
 
 # Gradio Interface
 interface = gr.Interface(
@@ -52,8 +64,8 @@ interface = gr.Interface(
         gr.Slider(label="Number of AIs", minimum=1, maximum=10, step=1)
     ],
     outputs=[
-        gr.JSON(label="Poll Results"),
-        gr.Textbox(label="AI Explanations")
+        gr.HTML(label="Poll Results"),  # Styled Output
+        gr.Textbox(label="Raw AI Explanations")
     ]
 )