Added code to handle java files

app.py

@@ -42,21 +42,58 @@ if uploaded_files:
             f.write(uploaded_file.getbuffer())
         st.success("File uploaded...")
 
+        # Check file type
+        _, file_extension = os.path.splitext(uploaded_file.name)
+
         st.success("Fetching list of functions...")
         file_path = f"./data/{uploaded_file.name}"
-        def extract_functions_from_file(file_path):
-            with open(file_path, "r") as file:
-                file_content = file.read()
-            
-            parsed_content = ast.parse(file_content)
-            functions = {}
-            
-            for node in ast.walk(parsed_content):
-                if isinstance(node, ast.FunctionDef):
-                    func_name = node.name
-                    func_body = ast.get_source_segment(file_content, node)
-                    functions[func_name] = func_body
-            
+        
+        def extract_functions_from_file(file_path, file_extension):
+
+            if file_extension == 'py':
+                with open(file_path, "r") as file:
+                    file_content = file.read()
+                
+                parsed_content = ast.parse(file_content)
+                functions = {}
+                
+                for node in ast.walk(parsed_content):
+                    if isinstance(node, ast.FunctionDef):
+                        func_name = node.name
+                        func_body = ast.get_source_segment(file_content, node)
+                        functions[func_name] = func_body
+                        
+            elif file_extension == 'java':
+                with open(file_path, 'r') as file:
+                    lines = file.readlines()
+                
+                functions = {}
+                inside_method = False
+                method_name = None
+                method_body = []
+                brace_count = 0
+                
+                method_signature_pattern = re.compile(r'((?:public|protected|private|static|\s)*)\s+[\w<>\[\]]+\s+(\w+)\s*\([^)]*\)\s*\{')
+                
+                for line in lines:
+                    if not inside_method:
+                        match = method_signature_pattern.search(line)
+                        if match:
+                            modifiers, method_name = match.groups()
+                            inside_method = True
+                            method_body.append(line)
+                            brace_count = line.count('{') - line.count('}')
+                    else:
+                        method_body.append(line)
+                        brace_count += line.count('{') - line.count('}')
+                        if brace_count == 0:
+                            inside_method = False
+                            functions[method_name] = ''.join(method_body)
+                            method_body = []
+                            
+                if 'main' in functions.keys():
+                    del(functions['main'])
+
             return functions
         
         functions = extract_functions_from_file(file_path)