Update app.py

app.py

@@ -3,11 +3,28 @@ import streamlit as st
 import requests
 import json
 import sqlite3
+import os
 from transformers import pipeline
 
 # Load AI model for test case generation
 test_case_generator = pipeline("text-generation", model="microsoft/CodeGPT-small-py")
 
+# Agar database file exist nahi karti toh create karo
+if not os.path.exists(DB_PATH):
+    conn = sqlite3.connect(DB_PATH)
+    cursor = conn.cursor()
+    cursor.execute('''
+        CREATE TABLE IF NOT EXISTS results (
+            id INTEGER PRIMARY KEY AUTOINCREMENT,
+            api_name TEXT,
+            test_status TEXT,
+            run_timestamp DATETIME DEFAULT CURRENT_TIMESTAMP
+        )
+    ''')
+    conn.commit()
+    conn.close()
+    
+
 def fetch_swagger_data(swagger_url):
     try:
         response = requests.get(swagger_url)
@@ -41,13 +58,14 @@ def store_test_results(test_case, result):
     conn.commit()
     conn.close()
 
+# Function to get past results
 def show_past_results():
-    conn = sqlite3.connect("test_results.db")
+    conn = sqlite3.connect(DB_PATH)
     cursor = conn.cursor()
     cursor.execute("SELECT * FROM results")
-    results = cursor.fetchall()
+    data = cursor.fetchall()
     conn.close()
-    return results
+    return data
 
 # Streamlit UI
 st.title("AI-Powered API Test Case Generator")