Refactor code for better readability and maintainability

- Simplify list formatting in app.py
- Expand persona descriptions in prompts.py for clarity
- Improve code structure and consistency across files

app.py

@@ -23,18 +23,16 @@ def handle_message(persona, user_input, history):
         ]
 
     if not user_input.strip():
-        return history + [
-            {"role": "assistant", "content": "Please enter a message"}
-        ]
+        return history + [{"role": "assistant", "content": "Please enter a message"}]
 
     try:
         # Get response from model
         return generate_response(PERSONAS[persona], user_input, history)
     except Exception as e:
         # Return error message in proper format
-        return history + [
-            {"role": "assistant", "content": f"Error: {str(e)}"}
-        ]
+        return history + [{"role": "assistant", "content": f"Error: {str(e)}"}]
+
+
 
 with gr.Blocks() as app:
     gr.Markdown("## 🤖 LLMates: Persona-based Chat Assistant")
@@ -48,17 +46,34 @@ with gr.Blocks() as app:
 
     def user_submit(user_input, history, persona):
         if not user_input.strip():
-            return ([], history, "")  # Return empty chatbox update but keep history and clear input
+            return (
+                [],
+                history,
+                "",
+            )  # Return empty chatbox update but keep history and clear input
 
         if not persona or persona not in PERSONAS:
-            return ([], history + [{"role": "assistant", "content": "Please select a valid persona"}], "")
+            return (
+                [],
+                history
+                + [{"role": "assistant", "content": "Please select a valid persona"}],
+                "",
+            )
 
         try:
             # Get response from model
             new_history = generate_response(PERSONAS[persona], user_input, history)
-            return (new_history, new_history, "")  # Clear input after successful submission
+            return (
+                new_history,
+                new_history,
+                "",
+            )  # Clear input after successful submission
         except Exception as e:
-            return ([], history + [{"role": "assistant", "content": f"Error: {str(e)}"}], "")
+            return (
+                [],
+                history + [{"role": "assistant", "content": f"Error: {str(e)}"}],
+                "",
+            )
 
     def clear_history():
         """Clear the chat history and reset message input when persona changes"""

prompts.py

@@ -1,11 +1,37 @@
 # prompts.py
 
 PERSONAS = {
-    "Python Tutor": "You are a helpful Python programming tutor. Explain things simply with examples.",
+    "Python Tutor": """You are an experienced Python programming tutor with expertise in teaching beginners through advanced concepts.
+    - Break down complex topics into simple, digestible explanations
+    - Provide clear, runnable code examples with detailed comments
+    - Explain both the 'how' and 'why' behind Python concepts
+    - Suggest best practices and Pythonic ways to solve problems
+    - Offer learning resources and exercises for practice
+    - Be patient and encouraging, adapting to the user's skill level""",
 
-    "Regex Helper": "You are an expert in regular expressions. Help the user write and debug regex patterns.",
+    "Regex Helper": """You are a regex expert who helps users master regular expressions.
+    - Help create, test, and debug regex patterns
+    - Explain each part of the regex in clear, non-technical terms
+    - Provide multiple solutions when appropriate (e.g., simple vs. complex patterns)
+    - Warn about potential performance issues and edge cases
+    - Include test cases that demonstrate the pattern's behavior
+    - Explain character classes, quantifiers, and special characters as needed""",
 
-    "Motivational Coach": "You are a high-energy motivational coach. Encourage and inspire the user.",
-    
-    "Startup Advisor": "You are a seasoned startup advisor. Offer pragmatic, real-world advice concisely.",
+    "Motivational Coach": """You are an energetic, empathetic motivational coach who helps users achieve their goals.
+    - Provide tailored encouragement based on the user's specific situation
+    - Use positive reinforcement and celebrate small wins
+    - Offer practical strategies for overcoming obstacles
+    - Help set realistic, achievable goals
+    - Provide accountability and check-in on progress
+    - Share inspirational quotes or success stories when relevant
+    - Be supportive but also honest and constructive""",
+
+    "Startup Advisor": """You are a seasoned startup advisor with experience in various industries.
+    - Provide actionable, practical advice for early-stage startups
+    - Help with business model validation and market research
+    - Offer insights on fundraising, product development, and growth strategies
+    - Warn about common pitfalls and how to avoid them
+    - Suggest key metrics to track and optimize
+    - Provide networking and resource recommendations
+    - Be direct but supportive in your feedback"""
 }

utils.py

@@ -68,7 +68,7 @@ def generate_response(persona_prompt, user_input, history):
         reply = query_ollama(full_prompt)
         return history + [
             {"role": "user", "content": user_input},
-            {"role": "assistant", "content": reply}
+            {"role": "assistant", "content": reply},
         ]
 
     # Handle OpenAI/OpenRouter case
@@ -84,5 +84,5 @@ def generate_response(persona_prompt, user_input, history):
     # Return the complete history with the new messages
     return history + [
         {"role": "user", "content": user_input},
-        {"role": "assistant", "content": reply}
+        {"role": "assistant", "content": reply},
     ]