Update app.py

app.py

@@ -1,6 +1,7 @@
 from transformers import AutoModelForCausalLM, AutoTokenizer
 import gradio as gr
 import torch
+import tempfile  # ✅ Import tempfile to create temp files
 
 # ✅ Load the fastest model on CPU
 model_name = "EleutherAI/pythia-70m"  # Fastest model for code review
@@ -23,21 +24,26 @@ def review_code(code_snippet):
     reviewed_code = tokenizer.decode(outputs[0], skip_special_tokens=True)
     print("✅ Generated Code:", reviewed_code)  # Debugging log
 
-    return reviewed_code
+    # ✅ Write reviewed code to a temporary file for download
+    temp_file_path = tempfile.NamedTemporaryFile(delete=False, suffix=".txt").name
+    with open(temp_file_path, "w") as temp_file:
+        temp_file.write(reviewed_code)
+
+    return reviewed_code, temp_file_path  # ✅ Return reviewed code & file path
 
 # ✅ Handle user input and return reviewed code
 def check_code(input_code):
-    reviewed_code = review_code(input_code)
-    return input_code, reviewed_code, reviewed_code  # Return all for UI & download
+    reviewed_code, file_path = review_code(input_code)
+    return input_code, reviewed_code, file_path  # ✅ Correctly return file path
 
-# ✅ Gradio UI with Side-by-Side Comparison & Download Option
+# ✅ Gradio UI with Side-by-Side Comparison & Fixed Download Option
 interface = gr.Interface(
     fn=check_code,
     inputs=gr.Textbox(label="Enter Python Code"),
     outputs=[
         gr.Textbox(label="Original Code", interactive=False),  # Left side
         gr.Textbox(label="Reviewed Code", interactive=False),  # Right side
-        gr.File(label="Download Reviewed Code")  # Download button
+        gr.File(label="Download Reviewed Code")  # ✅ Fixed Download Button
     ],
     title="🚀 AI Code Reviewer",
     description="📌 Enter Python code and get a reviewed version. Download the reviewed code as a file.",