Update app.py

app.py

@@ -2,12 +2,15 @@ import gradio as gr
 from openai import OpenAI
 
 def generate_solution(api_key, problem_statement):
+    # Check if inputs are provided
+    if not api_key or not problem_statement:
+        return "Please provide both an API key and a problem statement."
+    
     try:
         client = OpenAI(
             base_url="https://openrouter.ai/api/v1",
             api_key=api_key.strip(),
         )
-
         completion = client.chat.completions.create(
             model="open-r1/olympiccoder-7b:free",
             messages=[
@@ -23,7 +26,6 @@ def generate_solution(api_key, problem_statement):
             temperature=0.3,
             max_tokens=2048
         )
-
         response = completion.choices[0].message.content
         return format_response(response)
     
@@ -39,8 +41,11 @@ def format_response(response):
     for line in response.split('\n'):
         if line.strip().startswith('```'):
             if in_code and code_buffer:
-                # Fix: Separate string concatenation
-                code_content = '\n'.join(code_buffer[1:])  # Remove first line (language marker)
+                # Skip language marker if present
+                if len(code_buffer) > 0 and any(code_buffer[0].strip().lower() == lang for lang in ["python", "java", "c++", "cpp"]):
+                    code_content = '\n'.join(code_buffer[1:])
+                else:
+                    code_content = '\n'.join(code_buffer)
                 formatted.append(f"```python\n{code_content}\n```")
                 code_buffer = []
             in_code = not in_code
@@ -51,6 +56,14 @@ def format_response(response):
         else:
             explanation_buffer.append(line)
     
+    # Handle case where code block wasn't properly closed
+    if in_code and code_buffer:
+        if len(code_buffer) > 0 and any(code_buffer[0].strip().lower() == lang for lang in ["python", "java", "c++", "cpp"]):
+            code_content = '\n'.join(code_buffer[1:])
+        else:
+            code_content = '\n'.join(code_buffer)
+        formatted.append(f"```python\n{code_content}\n```")
+    
     if explanation_buffer:
         formatted.append("### Explanation\n" + '\n'.join(explanation_buffer))
         
@@ -64,7 +77,8 @@ with gr.Blocks(title="Competitive Programming Assistant") as app:
         api_key = gr.Textbox(
             label="OpenRouter API Key",
             type="password",
-            placeholder="Enter your API key here..."
+            placeholder="Enter your API key here...",
+            value=""  # Set default value to empty string
         )
     
     with gr.Row():
@@ -76,14 +90,14 @@ with gr.Blocks(title="Competitive Programming Assistant") as app:
     
     submit_btn = gr.Button("Generate Solution", variant="primary")
     
-    output = gr.Markdown()
+    output = gr.Markdown(label="Solution")
     
     submit_btn.click(
         fn=generate_solution,
         inputs=[api_key, problem_input],
         outputs=output
     )
-
+    
     gr.Examples(
         examples=[
             [
@@ -97,4 +111,4 @@ with gr.Blocks(title="Competitive Programming Assistant") as app:
     )
 
 if __name__ == "__main__":
-    app.launch(server_port=7860, share=False)
+    app.launch(server_port=7860, share=True)  # Changed share to True to make it accessible externally