Spaces:

thechaiexperiment
/

TeaRAG

Sleeping

App Files Files Community

thechaiexperiment commited on Jan 24

Commit

53a6059

verified ·

1 Parent(s): 92cfe6c

Update app.py

Browse files

Files changed (1) hide show

app.py +93 -74

app.py CHANGED Viewed

@@ -3,13 +3,12 @@ import pickle
 import os
 import re
 import numpy as np
 import torchvision
 import nltk
 import torch
 import pandas as pd
-import streamlit as st
 import requests
-import threading
 import zipfile
 import tempfile
 from PyPDF2 import PdfReader
@@ -36,13 +35,6 @@ from safetensors.torch import safe_open
 nltk.download('punkt_tab')
 app = FastAPI()
-with open("style.css") as f:
-    st.markdown(f"<style>{f.read()}</style>", unsafe_allow_html=True)
-# Streamlit interface
-def start_streamlit():
-    st.title("My Hugging Face App")
-    st.text_input("Ask a question:")
 app.add_middleware(
     CORSMiddleware,
@@ -511,9 +503,99 @@ def translate_en_to_ar(text):
         print(f"Error during English to Arabic translation: {e}")
         return None
-@app.get("/")
 async def root():
-    return {"message": "Welcome to the FastAPI application! Use the /health endpoint to check health, and /api/query for processing queries."}
 @app.get("/health")
 async def health_check():
@@ -662,66 +744,3 @@ if not init_success:
 if __name__ == "__main__":
     import uvicorn
     uvicorn.run(app, host="0.0.0.0", port=7860)
-# Chat API
-st.header("Chat API")
-chat_query = st.text_input("Enter your chat query:")
-language_code = st.selectbox("Select language", ["English", "Arabic"])
-if st.button("Submit Chat Query"):
-    response = requests.post(f"{BASE_URL}/api/chat", json={
-        "query": chat_query,
-        "language_code": 0 if language_code == "Arabic" else 1
-    })
-    if response.status_code == 200:
-        st.success("Response: " + response.json()['response'])
-    else:
-        st.error(f"Error: {response.status_code} - {response.text}")
-# Resources API
-st.header("Resources API")
-conditions = st.text_input("Enter medical conditions:")
-daily_symptoms = st.text_area("Enter daily symptoms:")
-count = st.number_input("Number of resources", min_value=1, max_value=10, value=5)
-if st.button("Submit Resources Query"):
-    response = requests.post(f"{BASE_URL}/api/resources", json={
-        "conditions": conditions,
-        "daily_symptoms": daily_symptoms,
-        "count": count
-    })
-    if response.status_code == 200:
-        resources = response.json()
-        st.write("Relevant Resources:")
-        for resource in resources:
-            st.markdown(f"- **{resource['title']}**: [Link]({resource['url']})")
-    else:
-        st.error(f"Error: {response.status_code} - {response.text}")
-# Recipes API
-st.header("Recipes API")
-recipe_conditions = st.text_input("Enter conditions for recipes:")
-recipe_symptoms = st.text_area("Enter symptoms for recipes:")
-recipe_count = st.number_input("Number of recipes", min_value=1, max_value=10, value=5)
-if st.button("Submit Recipes Query"):
-    response = requests.post(f"{BASE_URL}/api/recipes", json={
-        "conditions": recipe_conditions,
-        "daily_symptoms": recipe_symptoms,
-        "count": recipe_count
-    })
-    if response.status_code == 200:
-        recipes = response.json()['recipes']
-        st.write("Relevant Recipes:")
-        for recipe in recipes:
-            st.markdown(f"- **{recipe['title']}**: [Link]({recipe['url']})")
-    else:
-        st.error(f"Error: {response.status_code} - {response.text}")
-if __name__ == "__main__":
-    # Start FastAPI in a background thread
-    fastapi_thread = threading.Thread(target=run_fastapi)
-    fastapi_thread.start()
-    # Start Streamlit
-    start_streamlit()

 import os
 import re
 import numpy as np
+from fastapi.responses import HTMLResponse
 import torchvision
 import nltk
 import torch
 import pandas as pd
 import requests
 import zipfile
 import tempfile
 from PyPDF2 import PdfReader
 nltk.download('punkt_tab')
 app = FastAPI()
 app.add_middleware(
     CORSMiddleware,
         print(f"Error during English to Arabic translation: {e}")
         return None
+@app.get("/", response_class=HTMLResponse)
 async def root():
+    return """
+    <!DOCTYPE html>
+    <html>
+    <head>
+        <title>Simple User Interface</title>
+        <style>
+            body {
+                font-family: Arial, sans-serif;
+                margin: 20px;
+                background-color: #f8f9fa;
+            }
+            .container {
+                max-width: 600px;
+                margin: auto;
+                padding: 20px;
+                background: white;
+                border-radius: 8px;
+                box-shadow: 0px 4px 6px rgba(0, 0, 0, 0.1);
+            }
+            h1 {
+                color: #333;
+                text-align: center;
+            }
+            label {
+                display: block;
+                margin-bottom: 8px;
+                font-weight: bold;
+            }
+            input, select, button {
+                width: 100%;
+                padding: 10px;
+                margin-bottom: 20px;
+                border: 1px solid #ccc;
+                border-radius: 5px;
+            }
+            button {
+                background-color: #007bff;
+                color: white;
+                font-weight: bold;
+                cursor: pointer;
+            }
+            button:hover {
+                background-color: #0056b3;
+            }
+            .response {
+                font-size: 1rem;
+                color: #28a745;
+                font-weight: bold;
+                margin-top: 20px;
+            }
+        </style>
+    </head>
+    <body>
+        <div class="container">
+            <h1>AI Query Interface</h1>
+            <form id="queryForm">
+                <label for="query">Enter your query:</label>
+                <input type="text" id="query" name="query" placeholder="Type your question..." required>
+                <label for="language">Choose language:</label>
+                <select id="language" name="language">
+                    <option value="1">English</option>
+                    <option value="0">Arabic</option>
+                </select>
+                <button type="button" onclick="sendQuery()">Submit</button>
+            </form>
+            <div class="response" id="responseOutput"></div>
+        </div>
+        <script>
+            async function sendQuery() {
+                const query = document.getElementById('query').value;
+                const language = document.getElementById('language').value;
+                const response = await fetch('/api/chat', {
+                    method: 'POST',
+                    headers: {
+                        'Content-Type': 'application/json',
+                    },
+                    body: JSON.stringify({ query, language_code: parseInt(language) }),
+                });
+                const result = await response.json();
+                document.getElementById('responseOutput').innerText = result.response || 'No response';
+            }
+        </script>
+    </body>
+    </html>
+    """
 @app.get("/health")
 async def health_check():
 if __name__ == "__main__":
     import uvicorn
     uvicorn.run(app, host="0.0.0.0", port=7860)