Update app.py

app.py

@@ -1,6 +1,12 @@
 import gradio
 import subprocess
+import re
 
+def extract_class_name(java_code):
+  match = re.search(r'class\s+(\w+)\s*\{', java_code)
+    if match:
+        return match.group(1)
+    return None
 
 def run_command(command):
     process = subprocess.Popen(command, shell=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
@@ -32,14 +38,18 @@ def run_code(command):
             except subprocess.CalledProcessError as e:
                 result = str(e)
         elif lang == 'java':
+             class_name = extract_class_name(code)
             try:
-                with open('Main.java', 'w') as f:
+                 
+                with open(f'{class_name}.java', 'w') as f:
                     f.write(code)
-                compile_command = f"javac Main.java"
+                compile_command = f"javac {class_name}.java"
                 compile_process = subprocess.Popen(compile_command, shell=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
                 compile_output, compile_error = compile_process.communicate()
-                if compile_process.returncode == 0:
-                    run_command = f"java Main"
+                if not class_name:
+                    result ='Unable to extract class name from code'
+                elif compile_process.returncode == 0:
+                    run_command = f"java {class_name}"
                     run_process = subprocess.Popen(run_command, shell=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
                     run_output, run_error = run_process.communicate()
                     if run_process.returncode == 0: