Update app.py

app.py

@@ -39,26 +39,23 @@ def generate_code(prompt, task_type, language, max_tokens, temperature):
         else:  # Explain Code
             system_prompt = f"You are an expert {language} teacher. Explain the following code step by step:"
             full_prompt = f"{system_prompt}\n\n{prompt}\n\nExplanation:"
-        
-        messages = [
-            {"role": "user", "content": full_prompt},
-        ]
-        
+
         outputs = pipe(
-            messages,
+            full_prompt,
             max_new_tokens=int(max_tokens),
             temperature=temperature,
             do_sample=True,
             pad_token_id=pipe.tokenizer.eos_token_id
         )
         
-        generated_text = outputs[0]["generated_text"][-1]["content"] if isinstance(outputs[0]["generated_text"], list) else outputs[0]["generated_text"]
+        generated_text = outputs[0]["generated_text"]
         
         # Extract code if it's wrapped in code blocks
         code_match = re.search(r'```(?:\w+\n)?(.*?)```', generated_text, re.DOTALL)
         if code_match:
             code_output = code_match.group(1).strip()
         else:
+            # For "Explain Code" task, code_output should hold explanation text instead
             code_output = generated_text.strip()
         
         # Generate explanation based on the output
@@ -80,7 +77,6 @@ css = """
 .gradio-container {
     font-family: 'Segoe UI', Tahoma, Geneva, Verdana, sans-serif;
 }
-
 .header {
     text-align: center;
     background: linear-gradient(135deg, #667eea 0%, #764ba2 100%);
@@ -90,20 +86,17 @@ css = """
     margin-bottom: 2rem;
     box-shadow: 0 10px 30px rgba(0,0,0,0.2);
 }
-
 .header h1 {
     font-size: 2.5rem;
     font-weight: 700;
     margin: 0;
     text-shadow: 2px 2px 4px rgba(0,0,0,0.3);
 }
-
 .header p {
     font-size: 1.2rem;
     margin: 0.5rem 0 0 0;
     opacity: 0.9;
 }
-
 .custom-button {
     background: linear-gradient(135deg, #4facfe 0%, #00f2fe 100%);
     border: none;
@@ -112,12 +105,10 @@ css = """
     border-radius: 8px;
     transition: all 0.3s ease;
 }
-
 .custom-button:hover {
     transform: translateY(-2px);
     box-shadow: 0 5px 15px rgba(79, 172, 254, 0.4);
 }
-
 .footer {
     text-align: center;
     margin-top: 3rem;
@@ -127,12 +118,10 @@ css = """
     border-radius: 15px;
     box-shadow: 0 5px 15px rgba(0,0,0,0.1);
 }
-
 .footer h3 {
     margin: 0 0 1rem 0;
     font-size: 1.3rem;
 }
-
 .footer a {
     color: #74b9ff;
     text-decoration: none;
@@ -140,24 +129,20 @@ css = """
     font-weight: 500;
     transition: color 0.3s ease;
 }
-
 .footer a:hover {
     color: #0984e3;
 }
-
 .status-box {
     padding: 1rem;
     border-radius: 8px;
     margin: 1rem 0;
     font-weight: 500;
 }
-
 .code-output {
     background: #1e1e1e;
     border-radius: 8px;
     border: 1px solid #333;
 }
-
 .explanation-output {
     background: linear-gradient(135deg, #a8edea 0%, #fed6e3 100%);
     border-radius: 8px;
@@ -298,15 +283,15 @@ with gr.Blocks(css=css, title="AI Code Generator & Bug Fixer", theme=gr.themes.S
         outputs=[code_output, explanation_output]
     )
     
-    # Update code language based on selection
-    def update_code_language(lang):
-        return gr.Code(language=lang.lower())
-    
-    language.change(
-        fn=update_code_language,
-        inputs=language,
-        outputs=code_output
-    )
+    # Note: Dynamic update of the code block's language is tricky in Gradio.
+    # This is commented out as it won't update the existing gr.Code component properly.
+    # def update_code_language(lang):
+    #     return lang.lower()
+    # language.change(
+    #     fn=update_code_language,
+    #     inputs=language,
+    #     outputs=code_output.language
+    # )
 
 # Launch the app
 if __name__ == "__main__":
@@ -315,4 +300,4 @@ if __name__ == "__main__":
         server_port=7860,
         share=True,
         debug=True
-    )
+    )