Update run.py

run.py

@@ -96,8 +96,22 @@ def format_prompt(message, history):
     return prompt
 
 def response(prompt, history, temperature=0.9, max_new_tokens=500, top_p=0.95, repetition_penalty=1.0):
-    global global_url
-    print(f"Working with URL: {global_url}")  # You can use the URL here
+    global_url = ""  # Initialize URL variable
+    # JavaScript code to extract URL from the browser
+    js_code = """
+    <script>
+    function extractUrl() {
+        return window.location.href;
+    }
+    </script>
+    """
+    
+    # Extract URL using JavaScript
+    url_script = '<script>var url = extractUrl(); document.getElementById("url").innerText = url;</script>'
+    url_extracted = "<div id='url'></div>"  # Placeholder for URL extraction
+
+    print(f"Working with URL: {url_extracted}")
+    
     temperature = float(temperature)
     if temperature < 1e-2: temperature = 1e-2
     top_p = float(top_p)
@@ -130,51 +144,11 @@ def response(prompt, history, temperature=0.9, max_new_tokens=500, top_p=0.95, r
     now = str(datetime.now())
     save_to_sheet(now, prompt, output)
     yield output
+    
+gr.ChatInterface(
+    response,
+    chatbot=gr.Chatbot(value=[[None, "Herzlich willkommen! Ich bin Chätti ein KI-basiertes Studienassistenzsystem, das für jede Anfrage die am besten Studieninformationen empfiehlt.<br>Erzähle mir, was du gerne tust!"]], render_markdown=True),
+    title="German Studyhelper Chätti"
+).queue().launch(share=True)
 
-def js_code():
-    return """
-    <script>
-        function getUrl() {
-            const url = window.location.href;
-            const xhr = new XMLHttpRequest();
-            xhr.open("POST", "/submit_url", true);
-            xhr.setRequestHeader("Content-Type", "application/json");
-            xhr.onreadystatechange = function() {
-                if (xhr.readyState === 4 && xhr.status === 200) {
-                    console.log("URL submitted successfully");
-                }
-            };
-            xhr.send(JSON.stringify({ url: url }));
-        }
-        window.onload = getUrl;
-    </script>
-    """
-
-def submit_url(request):
-    global global_url
-    data = json.loads(request.body.decode("utf-8"))
-    url = data.get("url")
-    global_url = url  # Save the URL in the global variable
-    print(f"Received URL: {url}")
-    return {"status": "success"}
-
-# Define the Gradio interface
-def create_interface():
-    # Create a basic gr.Interface to include the JavaScript and handle URL submissions
-    with gr.Blocks() as app:
-        gr.HTML(js_code)  # Include the JavaScript in the app
-        gr.ChatInterface(
-            fn=response,
-            chatbot=gr.Chatbot(
-                value=[[None, "Herzlich willkommen! Ich bin Chätti ein KI-basiertes Studienassistenzsystem, das für jede Anfrage die am besten Studieninformationen empfiehlt.<br>Erzähle mir, was du gerne tust!"]],
-                render_markdown=True
-            ),
-            title="German Studyhelper Chätti"
-        ).launch(
-            share=True
-        )
-
-# Launch the Gradio app with URL capturing
-if __name__ == "__main__":
-    create_interface()
-    print("Interface up and running!")
+print("Interface up and running!")