Create app.py

app.py

@@ -0,0 +1,115 @@
+import os
+import httpx
+import gradio as gr
+
+# Defaulting to your Hugging Face space backend endpoint
+BACKEND_URL = os.getenv("BACKEND_URL", "https://gfathertech-mymodel.hf.space").rstrip("/")
+API_KEY = os.getenv("API_KEY", "")
+
+def get_headers():
+    headers = {}
+    if API_KEY:
+        headers["X-API-Key"] = API_KEY
+        headers["Authorization"] = f"Bearer {API_KEY}"
+    return headers
+
+# 1. File Processing Handler
+def process_file_api(file_obj, language, target_fmt):
+    if file_obj is None:
+        return {"error": "Please select a file to upload."}
+
+    url = f"{BACKEND_URL}/convert/file"
+    headers = get_headers()
+    
+    try:
+        with open(file_obj.name, "rb") as f:
+            files = {"file": (os.path.basename(file_obj.name), f, "application/octet-stream")}
+            data = {"language": language, "to": target_fmt}
+            
+            with httpx.Client(timeout=60.0) as client:
+                response = client.post(url, headers=headers, data=data, files=files)
+                
+            if response.status_code == 200:
+                return response.json()
+            return {"status": response.status_code, "detail": response.text}
+    except Exception as e:
+        return {"error": f"Connection failed: {str(e)}"}
+
+# 2. URL Processing Handler
+def process_url_api(url_input, language, target_fmt):
+    url = f"{BACKEND_URL}/convert/url"
+    headers = get_headers()
+    payload = {"url": url_input, "language": language, "to": target_fmt}
+
+    try:
+        with httpx.Client(timeout=60.0) as client:
+            response = client.post(url, headers=headers, json=payload)
+            
+        if response.status_code == 200:
+            return response.json()
+        return {"status": response.status_code, "detail": response.text}
+    except Exception as e:
+        return {"error": f"Connection failed: {str(e)}"}
+
+# 3. Chat Processing Handler
+def process_chat_api(message, history):
+    url = f"{BACKEND_URL}/chat"
+    headers = get_headers()
+    payload = {"message": message}
+
+    try:
+        with httpx.Client(timeout=30.0) as client:
+            response = client.post(url, headers=headers, json=payload)
+            
+        if response.status_code == 200:
+            res_data = response.json()
+            return res_data.get("response", str(res_data))
+        return f"Error {response.status_code}: {response.text}"
+    except Exception as e:
+        return f"Connection error: {str(e)}"
+
+# Build the Gradio UI
+with gr.Blocks(title="SOG-X Document Assistant", theme=gr.themes.Soft()) as demo:
+    gr.Markdown("# 📄 SOG-X Document & Assistant Suite")
+    
+    with gr.Tab("📁 Upload File"):
+        with gr.Row():
+            with gr.Column():
+                file_input = gr.File(label="Choose Document", file_types=[".pdf", ".docx", ".xlsx", ".pptx", ".txt"])
+                file_lang = gr.Textbox(value="en-US", label="Language Code", placeholder="e.g. en-US")
+                file_to = gr.Dropdown(choices=["pdf", "docx", "xlsx", "pptx"], value="pdf", label="Target Format")
+                file_btn = gr.Button("Convert File", variant="primary")
+            with gr.Column():
+                file_output = gr.JSON(label="Response Output")
+
+        file_btn.click(
+            process_file_api,
+            inputs=[file_input, file_lang, file_to],
+            outputs=file_output
+        )
+
+    with gr.Tab("🔗 Convert URL"):
+        with gr.Row():
+            with gr.Column():
+                url_input = gr.Textbox(placeholder="https://example.com/document.pdf", label="Document URL")
+                url_lang = gr.Textbox(value="en-US", label="Language Code", placeholder="e.g. en-US")
+                url_to = gr.Dropdown(choices=["pdf", "docx", "xlsx", "pptx"], value="pdf", label="Target Format")
+                url_btn = gr.Button("Convert URL", variant="primary")
+            with gr.Column():
+                url_output = gr.JSON(label="Response Output")
+
+        url_btn.click(
+            process_url_api,
+            inputs=[url_input, url_lang, url_to],
+            outputs=url_output
+        )
+
+    with gr.Tab("💬 Assistant Chat"):
+        gr.ChatInterface(
+            fn=process_chat_api,
+            textbox=gr.Textbox(placeholder="Ask a question about your documents...", container=False, scale=7),
+            title="SOG-X Chat Assistant"
+        )
+
+if __name__ == "__main__":
+    demo.launch()