Update app.py

app.py

@@ -1,64 +1,27 @@
-import os
-import tkinter as tk
-from tkinterdnd2 import TkinterDnD, DND_FILES
+import gradio as gr
 from transformers import AutoModelForCausalLM, AutoTokenizer
-import torch
 
-# Load the pre-trained model and tokenizer from Hugging Face
-model_name = "Salesforce/codegen-350M-mono"  # CodeGen model for code understanding
+# Load the model and tokenizer
+model_name = "codegen-350M-mono"
 model = AutoModelForCausalLM.from_pretrained(model_name)
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 
-# Function to explain Java code using the LLM model
 def explain_code(code):
     inputs = tokenizer(code, return_tensors="pt")
     outputs = model.generate(inputs['input_ids'], max_length=500)
     explanation = tokenizer.decode(outputs[0], skip_special_tokens=True)
     return explanation
 
-# Function to process the files from the folder and explain each Java file
-def process_folder(folder_path):
-    explanations = []
-    
-    # Walk through the directory and find all Java files
-    for filename in os.listdir(folder_path):
-        if filename.endswith(".java"):
-            file_path = os.path.join(folder_path, filename)
-            
-            # Read Java file contents
-            with open(file_path, 'r', encoding='utf-8') as file:
-                java_code = file.read()
-            
-            # Get explanation from the model
-            explanation = explain_code(java_code)
-            explanations.append(f"Explanation for {filename}:\n{explanation}\n")
-    
-    return explanations
+# Gradio interface for uploading files
+def process_uploaded_file(file):
+    code = file.read().decode("utf-8")
+    return explain_code(code)
 
-# Function to update the output area in the GUI
-def on_drop(event):
-    folder_path = event.data  # This gets the path of the dropped folder
-    if os.path.isdir(folder_path):
-        explanations = process_folder(folder_path)
-        output_text.delete(1.0, tk.END)  # Clear previous output
-        output_text.insert(tk.END, "\n\n".join(explanations))  # Display new output
+# Create the Gradio interface
+iface = gr.Interface(fn=process_uploaded_file, 
+                     inputs=gr.File(label="Upload Java File"), 
+                     outputs="text", 
+                     live=True)
 
-# Set up the GUI window
-root = TkinterDnD.Tk()
-root.title("Java Code Explainer")
-root.geometry("800x600")
-
-# Set up the text widget for displaying explanations
-output_text = tk.Text(root, wrap=tk.WORD, width=90, height=25)
-output_text.pack(padx=10, pady=10)
-
-# Set up the label for dragging and dropping the folder
-drop_label = tk.Label(root, text="Drag and drop a folder containing Java files here", width=80, height=2)
-drop_label.pack(padx=10, pady=10)
-
-# Register the drop area for folders
-root.drop_target_register(DND_FILES)
-root.dnd_bind('<<Drop>>', on_drop)
-
-# Start the GUI event loop
-root.mainloop()
+# Launch the interface
+iface.launch()