Update app.py

app.py

@@ -6,12 +6,13 @@ def generate_code(task, language="python", temperature=0.2):
     """Generate code based on task description"""
     examples = {
         "python": {
-            "reverse a string": "def reverse_string(s):\n    return s[::-1]\n\n# Example\nprint(reverse_string('hello'))  # Output: olleh",
+            "reverse a string": "def reverse_string(s):\n    return s[::-1]",
+             "(reverse_string('hello'))  # Output: HELLO",
             "check if prime": "def is_prime(n):\n    if n < 2:\n        return False\n    for i in range(2, int(n**0.5) + 1):\n        if n % i == 0:\n            return False\n    return True",
             "default": "# Python function\ndef example_function():\n    pass"
         },
         "javascript": {
-            "reverse a string": "function reverseString(str) {\n    return str.split('').reverse().join('');\n}\n\n// Example\nconsole.log(reverseString('hello')); // Output: olleh",
+            "reverse a string": "function reverseString(str) {\n    return str.split('').reverse().join('');\n}",
             "default": "// JavaScript function\nfunction exampleFunction() {\n  pass  // Implementation here\n}"
         }
     }
@@ -23,12 +24,10 @@ def generate_code(task, language="python", temperature=0.2):
     return lang_examples["default"]
 
 def explain_code(code):
-    """Explain what the code does"""
-    return "This code implements a common programming task. It takes input, processes it according to the defined logic, and returns the expected output."
+        return "This code processes input and returns the expected output. This code implements a common programming task. It takes input, processes it according to the defined logic, and returns the expected output. (Detailed AI explanation will go here.)"
 
 def fix_code(code, error=""):
-    """Fix code with optional error message"""
-    return "# Fixed version of the code\n" + code.replace("TODO", "# Implementation fixed")
+        return "# Fixed version of the code\n" + code.replace("TODO", "# Implementation fixed")
 
 def chat_with_ai(message, history):
     """Chat interface for general programming questions"""
@@ -44,19 +43,45 @@ def chat_with_ai(message, history):
     return responses["default"]
 
 
-# ----------------------------------------------------
-# UI — Gradio Blocks Interface
-# ----------------------------------------------------
+# ---------------- UI Interface ----------------
 with gr.Blocks(title="SynapseAI Programming Assistant") as app:
     gr.Markdown("# ⚡ SynapseAI — Your Coding Partner")
 
-    with gr.Tab("Generate Code"):
+# Generate Code Tab    
+with gr.Tab("Generate Code"):
         task = gr.Textbox(label="Describe what you want to build")
         lang = gr.Dropdown(["python", "javascript", "cpp", "java"], value="python", label="Language")
         temp = gr.Slider(0.1, 1.0, value=0.2, label="Creativity")
         generate = gr.Button("Generate Code")
         output_code = gr.Code(language="python", label="Output")
-    
+   
+generate_btn.click(generate_code, [task, lang, temp], code_output)
+
+    # Explain Code Tab
+    with gr.Tab("Explain Code"):
+        code_input = gr.Textbox(label="Paste code here", lines=10)
+        explain_btn = gr.Button("Explain Code")
+        explain_output = gr.Textbox(label="Explanation", lines=10)
+        explain_btn.click(explain_code, code_input, explain_output)
+
+    # Fix Code Tab
+    with gr.Tab("Fix Code"):
+        broken_code = gr.Textbox(label="Broken code", lines=10)
+        error_msg = gr.Textbox(label="Error message (optional)")
+        fix_btn = gr.Button("Fix Code")
+        fixed_output = gr.Code(label="Fixed Code")
+        fix_btn.click(fix_code, [broken_code, error_msg], fixed_output)
+
+    # Chat Tab
+    with gr.Tab("Chat"):
+        chatbot = gr.Chatbot(label="Programming Assistant")
+        chat_input = gr.Textbox(label="Ask anything...")
+        send_chat = gr.Button("Send")
+        clear_chat = gr.Button("Clear Chat")
+
+        send_chat.click(chat_ai, [chat_input, chatbot], chatbot)
+        clear_chat.click(lambda: None, None, chatbot)
+ 
         with gr.Tab("Generate Code"):
         with gr.Row():
             with gr.Column():
@@ -149,4 +174,4 @@ with gr.Blocks(title="SynapseAI Programming Assistant") as app:
 
 # For Hugging Face Spaces
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()