Update app.py

app.py

@@ -1,64 +1,68 @@
-import gradio as gr
-import requests
-from getStatus import run  # Assuming this module contains the run function
-
-def getURL(URL):
-    if URL.startswith("http://") or URL.startswith("https://"):
-        websiteUrl = URL
-    else:
-        websiteUrl = "http://" + URL
-    return websiteUrl
-
-def setHeaders(websiteUrl):
-    try:
-        URL = getURL(websiteUrl)
-        web_response = requests.get(URL, timeout=5)
-        user_agent = web_response.request.headers['User-Agent']
-        headers = {'User-Agent': user_agent}
-        return headers
-    except requests.RequestException as e:
-        print(f"Error fetching headers: {e}")
-        return None
-
-def websiteStatus(websiteUrl, user_agent_choice):
-    try:
-        URL = getURL(websiteUrl)
-        
-        if user_agent_choice == "Your Browser Headers":
-            headers = setHeaders(websiteUrl)
-        else:
-            headers = {'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/58.0.3029.110 Safari/537.3'}
-        
-        web_response = requests.get(URL, headers=headers)
-        web_status = str(web_response.status_code)
-        code, webstatus, status, morestatus = run(web_status)
-        print(f"Processed Output: {code}, {webstatus}, {status}, {morestatus}")  # Debug print
-        return code, webstatus, status, morestatus
-    
-    except requests.ConnectionError:
-        error = f"Failed to connect to {websiteUrl}"
-        return error, error, error, error
-    except requests.Timeout:
-        error = f"Request to {websiteUrl} timed out"
-        return error, error, error, error
-    except requests.RequestException as e:
-        error = f"An error occurred: {e}"
-        return error, error, error, error
-
-# Create a Gradio interface
-app = gr.Interface(
-    fn=websiteStatus,
-    inputs=[
-        gr.Textbox(label="Enter URL", placeholder="https://google.com", type="text"),
-        gr.Radio(["Your Browser Headers", "Our Server Headers"], label="Select User Agent")
-    ],
-    outputs=[
-        gr.Textbox(label="Code", type="text"),
-        gr.Textbox(label="Server/Website Status", type="text"),
-        gr.Textbox(label="Code Status", type="text"),
-        gr.Textbox(label="More Code Status Information", type="text")
-    ]
-)
-
-if __name__ == "__main__":
-    app.launch()
+import gradio as gr
+import requests
+from getStatus import run  # Assuming this module contains the run function
+
+def getURL(URL):
+    if URL.startswith("http://") or URL.startswith("https://"):
+        websiteUrl = URL
+    else:
+        websiteUrl = "http://" + URL
+    return websiteUrl
+
+def setHeaders(websiteUrl):
+    try:
+        URL = getURL(websiteUrl)
+        web_response = requests.get(URL, timeout=5)
+        user_agent = web_response.request.headers['User-Agent']
+        headers = {'User-Agent': user_agent}
+        return headers
+    except requests.RequestException as e:
+        print(f"Error fetching headers: {e}")
+        return None
+
+def websiteStatus(websiteUrl, user_agent_choice):
+    try:
+        URL = getURL(websiteUrl)
+        
+        if user_agent_choice == "Your Browser Headers":
+            headers = setHeaders(websiteUrl)
+        else:
+            headers = {'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/58.0.3029.110 Safari/537.3'}
+        
+        web_response = requests.get(URL, headers=headers)
+        web_status = str(web_response.status_code)
+        code, webstatus, status, morestatus = run(web_status)
+        print(f"Processed Output: {code}, {webstatus}, {status}, {morestatus}")  # Debug print
+        return code, webstatus, status, morestatus
+    
+    except requests.ConnectionError:
+        error = f"Failed to connect to {websiteUrl}"
+        return error, error, error, error
+    except requests.Timeout:
+        error = f"Request to {websiteUrl} timed out"
+        return error, error, error, error
+    except requests.RequestException as e:
+        error = f"An error occurred: {e}"
+        return error, error, error, error
+
+# Create a Gradio interface
+app = gr.Interface(
+    fn=websiteStatus,
+    inputs=[
+        gr.Textbox(label="Enter URL", placeholder="https://google.com", type="text"),
+        gr.Radio(["Your Browser Headers", "Our Server Headers"], label="Select User Agent")
+    ],
+    outputs=[
+        gr.Textbox(label="Code", type="text"),
+        gr.Textbox(label="Server/Website Status", type="text"),
+        gr.Textbox(label="Code Status", type="text"),
+        gr.Textbox(label="More Code Status Information", type="text")
+    ]
+    title="Website HTTP Status Checker<br> by <a href='https://nayankasturi.eu.org'>Nayan Kasturi</a> aka Raanna.<br> Checkout my <a href='https://github.com/raannakasturi'>Github</a> for more projects and contact info.",
+    description="This app scans the website for HTTP statuses.<br> Licenced under <a href='https://creativecommons.org/licenses/by-nc-sa/4.0/'>cc-by-nc-sa-4.0</a>",
+    api_name="get",
+    concurrency_limit=10
+)
+
+if __name__ == "__main__":
+    app.launch()