Update app.py

app.py

@@ -213,7 +213,7 @@ class TestCaseGenerator:
 
     def setup_streamlit_ui(self):
         st.title("Auto Test Case Generation using LLM")
-        uploaded_files = st.file_uploader("Upload a python or Java file", type=[".py", "java"], accept_multiple_files=True)
+        uploaded_files = st.file_uploader("Upload a python or Java file", type=["py", "java"], accept_multiple_files=True)
         if uploaded_files:
             for uploaded_file in uploaded_files:
                 self.process_uploaded_file(uploaded_file)
@@ -223,11 +223,11 @@ class TestCaseGenerator:
             f.write(uploaded_file.getbuffer())
         st.success("File uploaded...")
         _, file_extension = os.path.splitext(uploaded_file.name)
-        print(file_extension)
         st.success("Fetching list of functions...")
         file_path = f"./data/{uploaded_file.name}"
         self.extract_functions_from_file(file_path, file_extension)
         st.write(self.list_of_functions)
+      
 
     def extract_functions_from_file(self, file_path, file_extension):
         if file_extension == '.py':
@@ -294,16 +294,12 @@ class TestCaseGenerator:
             return
 
         snippet = self.functions[func]
-        _ , file_extension = os.path.splitext(uploaded_file.name)
-        lang = file_extension[1:]
        
         prompt = f"""1. You will be provided with the following inputs:
                            <function_code>
                            {snippet}
                            </function_code>
 
-                           <language>{lang}</language> 
-
                     2. Analyze the provided function:
                        - Identify the function name, parameters, and return type
                        - Determine the main logic and branches in the function
@@ -355,6 +351,7 @@ class TestCaseGenerator:
 
     def run(self):
         self.setup_streamlit_ui()
+        
         if "messages" not in st.session_state:
             st.session_state.messages = []