Updated

app.py

@@ -1,132 +1,32 @@
-import os
-import requests
-import gradio as gr # Or whatever UI you use
-from fastapi import Body
+from fastapi import FastAPI
+import gradio as gr
+from pydantic import BaseModel
 
-# 1. Pull your secret token
-HF_TOKEN = os.environ.get("HF_TOKEN")
-DB_HF_TOKEN = os.environ.get("DB_Write_Token")
+app = FastAPI()
 
-# 2. Define the full URL to your private FastAPI endpoint
-# Note: Use the direct .hf.space URL
-PRIVATE_API_URL = "https://rr19tech-plnb-demo.hf.space/get_content"
-FEEDBACK_API_URL = "https://rr19tech-plnb-demo.hf.space/add_user_comment"
-'''
-def call_private_api(user_input):
-    # Pass the token in the Authorization header
-    headers = {
-        "Authorization": f"Bearer {HF_TOKEN}",
-        #"Content-Type": "text/plain"
-        "Content-Type": "application/json"
-    }
+# 1. Define the incoming JSON structure
+class ApiInput(BaseModel):
+    user_string: str  # The key name you want to use in your JSON payload
 
-    print(f"User input is: {user_input}")
+# 2. Setup your POST route
+@app.post("/call_private_api")
+def call_private_api(payload: ApiInput):
+    # Extract the string value from the request
+    input_text = payload.user_string
+    
+    # Run your custom processing logic here
+    processed_result = f"Received and processed: {input_text}"
     
-    # We take that string and put it into a GET request
-    try:
-        response = requests.get(
-            PRIVATE_API_URL, 
-            headers=headers, 
-            params={"text": user_input},
-            timeout=300,
-        )
-        content_type = response.headers.get("content-type")
-        print(f"DEBUG: Status {response.status_code} | Content-Type: {content_type}")
-        
-        if response.status_code == 200:
-            #return response.json()
-            return response.text # workaround
-        else:
-            return {"error": f"Status {response.status_code}", "detail": response.text}
-    except Exception as e:
-        return {"error": f"Connection failed: {str(e)}"}
+    return {"result": processed_result}
 
-# Simple UI to test it
-demo = gr.Interface(fn=call_private_api, inputs="text", outputs="json").launch()
-#demo.queue().launch()
-'''
+# 3. Mount your standard Gradio interface
+with gr.Blocks() as demo:
+    gr.Markdown("# My Space Interface")
 
-# Pass the token in the Authorization header
+app = gr.mount_gradio_app(app, demo, path="/")
 
-#changing this to a server method
-from gradio import Server
-#from fastapi.middleware.cors import CORSMiddleware
 
-# Initialize the server
-app = Server()
-'''
-app.add_middleware(
-    CORSMiddleware,
-    # Replace with the EXACT domain where your JS code is running (include http/https)
-    # DO NOT USE ["*"] here, as browsers block credentials paired with wildcards
-    allow_origins=["http://localhost:8000", "https://your-frontend-app.com"],
-    allow_credentials=True,  # Matches the incoming JS request flag
-    allow_methods=["*"],
-    allow_headers=["*"],
-)
-'''
-# Define Method 1: Triggers when client calls /gradio_api/call/update_data
-#@app.api(name="call_private_api")
-@app.post("/call_private_api")
-async def call_private_api(user_input: str) -> dict:
-#def call_private_api(user_input: str = Body(..., embed=True)) -> dict:
 
-    headers = {
-        "Authorization": f"Bearer {HF_TOKEN}",
-        #"Content-Type": "text/plain"
-        "Content-Type": "application/json"
-    }
-    print(f"User input is: {user_input}")
-    
-    # We take that string and put it into a GET request
-    try:
-        response = requests.get(
-            PRIVATE_API_URL, 
-            headers=headers, 
-            params={"text": user_input},
-            timeout=300,
-        )
-        content_type = response.headers.get("content-type")
-        print(f"DEBUG: Status {response.status_code} | Content-Type: {content_type}")
-        
-        if response.status_code == 200:
-            return response.json()
-            #return response.text # workaround
-        else:
-            return {"error": f"Status {response.status_code}", "detail": response.text}
-    except Exception as e:
-        return {"error": f"Connection failed: {str(e)}"}
 
-# Define Method 2: User feedback
-
-#@app.api(name="user_feedback")
-@app.post("/user_feedback")
-async def process_user_feedback(user_input: str) -> dict:
-
-    headers = {
-        "Authorization": f"Bearer {DB_HF_TOKEN}",
-        #"Content-Type": "text/plain"
-        "Content-Type": "application/json"
-    }
-    print(f"User feedback input is : {user_input}")
-    try:
-        response = requests.get(
-            FEEDBACK_API_URL, 
-            headers=headers, 
-            params={"text": user_input},
-            timeout=300,
-        )
-        content_type = response.headers.get("content-type")
-        print(f"DEBUG: Status {response.status_code} | Content-Type: {content_type}")
-        
-        if response.status_code == 200:
-            return response.json()
-            #return response.text # workaround
-        else:
-            return {"error": f"Status {response.status_code}", "detail": response.text}
-    except Exception as e:
-        return {"error": f"Connection failed: {str(e)}"}
-    
 
-app.launch()