Update app.py

app.py

@@ -1,12 +1,12 @@
 import gradio
 import subprocess
-# import re
+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 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)
@@ -38,8 +38,8 @@ def run_code(command):
             except subprocess.CalledProcessError as e:
                 result = str(e)
         elif lang == 'java':
-            # class_name = extract_class_name(code)
-            class_name = 'main'
+            class_name = extract_class_name(code)
+            # class_name = 'main'
             try:
                  
                 with open(f'{class_name}.java', 'w') as f: