Update backend.py

backend/backend.py

@@ -7,6 +7,7 @@ import openai
 from dotenv import load_dotenv
 from easycompletion import openai_text_call
 from fastapi import FastAPI, HTTPException
+from fastapi import Request
 from fastapi.middleware.gzip import GZipMiddleware
 from fastapi.testclient import TestClient
 from gradio import Interface, TabbedInterface, components, mount_gradio_app
@@ -15,30 +16,10 @@ from pygments import highlight
 from pygments.formatters import html
 from pygments.lexers import get_lexer_by_name
 from sqlalchemy import JSON, Column, Integer, String, create_engine
+from fastapi.responses import FileResponse, JSONResponse
 from sqlalchemy.ext.declarative import declarative_base
 from sqlalchemy.orm import Session, sessionmaker
 from uvicorn import Config, Server
-from fuzzywuzzy import fuzz
-
-COMMANDS = {
-    "scroll_up": "acc:UP",
-    "scroll_down": "acc:DOWN",
-    "click_first": "acc:CLICK FIRST",
-    "click": "acc:CLICK"
-}
-
-def analyze_command(response):
-    response_lower = response.lower()
-    
-    for command_type, command_text in COMMANDS.items():
-        if fuzz.partial_ratio(response_lower, command_text.lower()) >= 90:
-            return command_type
-    
-    return None
-
-def generate_command_response(command_type):
-    return COMMANDS.get(command_type, None)
-
 
 LANGS = [
     "gpt-3.5-turbo",
@@ -247,33 +228,12 @@ async def saveurl(item: SaveURLItem):
 
 def assist_interface(uid, prompt, gpt_version):
     client = TestClient(app)
-    
-    modified_prompt = "If this text is asking you to either click on something, scroll up or down, or click on the first, return either acc:UP, acc:DOWN, acc:CLICK FIRST, or acc:CLICK. If it isn't, answer the question as normal. If it's asking you to click, click first, or scroll up or down, then return only acc:UP, acc:DOWN, acc:CLICK FIRST, or acc:CLICK, according to the command. Don't show anything before or after the command, nothing like 'Sure, here's the command you requested' just return the response. Here's the text:" + prompt
-    
+
     response = client.post(
         "/assist",
-        json={"uid": uid, "prompt": modified_prompt, "version": gpt_version},
+        json={"uid": uid, "prompt": prompt, "version": gpt_version},
     )
-    
-    response_data = json.loads(response.text)
-    generated_text = response_data["text"]
-    
-    command_type = analyze_command(generated_text)
-    
-    if command_type:
-        command_response = generate_command_response(command_type)
-        add_command_response = client.post(
-            "/add_command",
-            json={"uid": uid, "command": command_response},
-        )
-        
-        if add_command_response.status_code == 200:
-            return add_command_response.json()
-        else:
-            return {"error": "Error adding command to the queue"}
-    else:
-        return generate_html_response_from_openai(response.text)
-
+    return generate_html_response_from_openai(response.text)
 
 
 def get_user_interface(uid):
@@ -346,7 +306,6 @@ def register_interface(openai_key):
     )
     return response.json()
 
-
 def get_register_interface():
     def wrapper(openai_key):
         result = register_interface(openai_key)
@@ -363,13 +322,11 @@ def get_register_interface():
         description="Register a new user by entering an OpenAI key.",
     )
 
-
 def get_history_interface(uid):
     client = TestClient(app)
     response = client.get(f"/get_history/{uid}")
     return response.json()
 
-
 def get_history_gradio_interface():
     return Interface(
         fn=get_history_interface,
@@ -379,7 +336,6 @@ def get_history_gradio_interface():
         description="Get the history of questions and answers for a given user.",
     )
 
-
 def add_command_interface(uid, command):
     client = TestClient(app)
     response = client.post(
@@ -388,6 +344,49 @@ def add_command_interface(uid, command):
     )
     return response.json()
 
+@app.get("/.well-known/ai-plugin.json")
+async def plugin_manifest(request: Request):
+  host = request.headers['host']
+  with open(".well-known/ai-plugin.json") as f:
+    text = f.read().replace("PLUGIN_HOSTNAME", "https://posix4e-puppet.hf.space/")
+  return JSONResponse(content=json.loads(text))
+
+
+
+@app.get("/openapi.yaml")
+async def openai_yaml(request: Request):
+  host = request.headers['host']
+  with open(".well-known/openapi.yaml") as f:
+    text = f.read().replace("PLUGIN_HOSTNAME","https://posix4e-puppet.hf.space/")
+  return JSONResponse(content=json.loads(text))
+
+
+
+@app.get("/detectcommand/{command}")
+async def get_command(command: str, item: AssistItem):
+  db: Session = SessionLocal()
+  user = db.query(User).filter(User.uid == item.uid).first()
+  if not user:
+        raise HTTPException(status_code=400, detail="Invalid uid")
+  openai.api_key = user.openai_key
+  response = openai_text_call(item.prompt, model=item.version)
+  return JSONResponse(content= response, status_code=200)
+
+@app.get("/logo.png")
+async def plugin_logo():
+  return FileResponse('/.well-known/logo.jpeg')
+
+def get_add_command_interface():
+    return Interface(
+        fn=add_command_interface,
+        inputs=[
+            components.Textbox(label="UID", type="text"),
+            components.Textbox(label="Command", type="text"),
+        ],
+        outputs="json",
+        title="Add Command",
+        description="Add a new command for a given user.",
+    )
 
 app = mount_gradio_app(
     app,
@@ -397,6 +396,7 @@ app = mount_gradio_app(
             get_db_interface(),
             get_register_interface(),
             get_history_gradio_interface(),
+            get_add_command_interface(),
         ]
     ),
     path="/",