Update app.py

app.py

@@ -2,8 +2,9 @@ import gradio as gr
 import os
 from openai import OpenAI
 import time
+import markdown
 
-def solve_competitive_problem(problem_statement, language_choice, api_key):
+def solve_competitive_problem(problem_statement, language_choice, api_key, progress=gr.Progress()):
     """
     Generate a solution for a competitive programming problem
     
@@ -11,6 +12,7 @@ def solve_competitive_problem(problem_statement, language_choice, api_key):
         problem_statement (str): The problem statement
         language_choice (str): Programming language for the solution
         api_key (str): OpenRouter API key
+        progress: Gradio progress tracker
     
     Returns:
         str: Step-by-step solution with code
@@ -22,13 +24,17 @@ def solve_competitive_problem(problem_statement, language_choice, api_key):
         return "Error: Please provide a problem statement."
     
     try:
+        progress(0.1, "Initializing...")
+        
         # Initialize OpenAI client with OpenRouter base URL
         client = OpenAI(
             base_url="https://openrouter.ai/api/v1",
             api_key=api_key,
         )
         
-        # Create prompt with language preference
+        progress(0.3, "Creating prompt...")
+        
+        # Create a more detailed prompt with language preference
         prompt = f"""
 You are an expert competitive programmer. Analyze the following problem and provide a step-by-step solution with explanations and code in {language_choice}.
 
@@ -36,13 +42,43 @@ Problem:
 {problem_statement}
 
 Your response should include:
-1. Problem Analysis: Break down the problem statement and identify the key challenges
-2. Approach: Describe your approach to solve the problem and why it's optimal
-3. Algorithm: Explain the algorithm you'll use with time and space complexity analysis
-4. Implementation: Write clean, efficient, and well-commented {language_choice} code
-5. Testing: Explain how to test the solution with example inputs/outputs
+
+1. Problem Analysis:
+   - Clear restatement of the problem in your own words
+   - Identification of input/output formats
+   - Key constraints and edge cases to consider
+   - Time and space complexity requirements
+
+2. Approach:
+   - High-level strategy to solve the problem
+   - Why this approach is optimal compared to alternatives
+   - Any mathematical insights or observations
+   - Data structures that will be helpful
+
+3. Algorithm:
+   - Detailed step-by-step breakdown of the algorithm
+   - Clear explanation of the logic behind each step
+   - Time complexity analysis with justification
+   - Space complexity analysis with justification
+   - Any optimizations made to improve performance
+
+4. Implementation:
+   - Clean, efficient, and well-commented {language_choice} code
+   - Proper variable naming and code organization
+   - Error handling and edge case management
+   - Optimized for both readability and performance
+
+5. Testing:
+   - Example test cases with expected outputs
+   - Edge case testing scenarios
+   - Explanation of how to verify correctness
+   - Potential areas where the solution might need improvement
+
+Format your answer with clear headings and subheadings. Use markdown formatting for better readability.
         """
         
+        progress(0.5, "Generating solution...")
+        
         # Call the model
         completion = client.chat.completions.create(
             extra_headers={
@@ -60,7 +96,11 @@ Your response should include:
             stream=False
         )
         
+        progress(0.9, "Processing response...")
+        
         solution = completion.choices[0].message.content
+        
+        progress(1.0, "Complete!")
         return solution
         
     except Exception as e:
@@ -121,6 +161,15 @@ with gr.Blocks(title="Competitive Programming Assistant", theme=gr.themes.Soft()
         - Include the entire problem, including input/output formats and constraints
         - Make sure to include example inputs and outputs
         - For complex problems, consider adding clarifying notes
+        
+        ### Solution Format:
+        
+        Your solution will include:
+        1. **Problem Analysis** - Breaking down the problem into manageable components
+        2. **Approach** - Strategic methodology with mathematical insights
+        3. **Algorithm** - Detailed step-by-step procedure with complexity analysis
+        4. **Implementation** - Clean, commented code in your chosen language
+        5. **Testing** - Example test cases and verification methods
         """)
     
     # Handle form submission