Update app.py

app.py

@@ -1,49 +1,37 @@
 import streamlit as st
-from testing import estimate_testing_efforts, perform_all_tests, generate_test_report
-
-# Step 1: Exchange API Key for Access Token
-apikey = "6NV9V90_cdLLarB6FhcN5lQh2JVYf50mUfQ8C4KHNby3"
-auth_url = "https://iam.cloud.ibm.com/identity/token"
-
-# Fetch Access Token
-access_token = generate_text.get_access_token(apikey, auth_url)
-
-if not access_token:
-    st.error("Failed to retrieve access token.")
-    st.stop()
-
-# Title of the Streamlit app
-st.title("IBM Watson Text Generation and Testing Automation")
-
-# User input for the text generation prompt
-user_input = st.text_area("Enter your prompt:", value="You are Granite Chat, an AI language model developed by IBM. You are a cautious assistant.")
-
-# Text generation
-if st.button("Generate Text"):
-    generated_text = generate_text.generate_text(user_input, access_token)
-    if generated_text:
-        st.write("**IBM Watson Output:**")
-        st.write(generated_text)
-    else:
-        st.error("Failed to generate text.")
-
-# URL input for testing estimation
-url_input = st.text_input("Enter the URL of the website or web app for testing:")
-
-# Testing estimation
-if st.button("Estimate Testing Efforts"):
-    if url_input:
-        efforts = estimate_testing_efforts(url_input)
-        st.write("**Testing Efforts Estimation:**")
-        st.write(efforts)
-    else:
-        st.error("Please enter a URL.")
-
-# Perform tests and generate report
-if st.button("Perform Tests and Generate Report"):
-    if url_input:
-        test_report = perform_all_tests(url_input)
-        st.write("**Test Report:**")
-        st.write(test_report)
-    else:
-        st.error("Please enter a URL.")
+from estimation import estimate_testing_resources
+from automation import generate_automation_scripts
+from critical_functionality import test_critical_functionality
+from regression_testing import run_regression_tests
+from test_report import generate_test_report
+from sanity_check import perform_sanity_check
+from performance_testing import perform_performance_test
+
+def main():
+    st.title("Automated Testing Tool")
+
+    url = st.text_input("Enter the URL to be tested")
+    
+    if st.button("Estimate Testing Resources"):
+        st.write(estimate_testing_resources(url))
+    
+    if st.button("Generate Automation Test Scripts"):
+        st.write(generate_automation_scripts(url))
+    
+    if st.button("Test Critical Functionalities"):
+        st.write(test_critical_functionality(url))
+    
+    if st.button("Run Regression Tests"):
+        st.write(run_regression_tests(url))
+    
+    if st.button("Generate Test Report"):
+        st.write(generate_test_report(url))
+    
+    if st.button("Perform Sanity Check"):
+        st.write(perform_sanity_check(url))
+    
+    if st.button("Perform Performance Test"):
+        st.write(perform_performance_test(url))
+
+if __name__ == "__main__":
+    main()