Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -1,5 +1,7 @@
|
|
|
|
|
|
|
|
| 1 |
import gradio as gr
|
| 2 |
-
|
| 3 |
import requests
|
| 4 |
import json
|
| 5 |
import os
|
|
@@ -18,7 +20,7 @@ class RaindropSearchBot:
|
|
| 18 |
def __init__(self):
|
| 19 |
self.openai_api_key = OPENAI_API_KEY
|
| 20 |
self.raindrop_api_token = RAINDROP_TOKEN
|
| 21 |
-
|
| 22 |
|
| 23 |
def generate_search_query(self, user_request: str) -> str:
|
| 24 |
"""Convert user request to a tailored search query using OpenAI."""
|
|
@@ -31,15 +33,13 @@ class RaindropSearchBot:
|
|
| 31 |
Format the search query with relevant keywords and operators if needed.
|
| 32 |
"""
|
| 33 |
|
| 34 |
-
response =
|
| 35 |
-
model="gpt-
|
| 36 |
-
messages=[
|
| 37 |
-
|
| 38 |
-
|
| 39 |
-
]
|
| 40 |
)
|
| 41 |
-
|
| 42 |
-
return response.choices[0].message.content.strip()
|
| 43 |
|
| 44 |
def search_raindrop(self, search_query: str) -> List[Dict]:
|
| 45 |
"""Search Raindrop.io with the generated query."""
|
|
@@ -122,7 +122,8 @@ with gr.Blocks(title="Raindrop.io Link Search Assistant", theme=gr.themes.Soft()
|
|
| 122 |
with gr.Row():
|
| 123 |
output_text = gr.Textbox(
|
| 124 |
label="Search Results",
|
| 125 |
-
lines=10
|
|
|
|
| 126 |
)
|
| 127 |
|
| 128 |
search_button.click(
|
|
@@ -142,4 +143,4 @@ with gr.Blocks(title="Raindrop.io Link Search Assistant", theme=gr.themes.Soft()
|
|
| 142 |
|
| 143 |
# Launch the interface
|
| 144 |
if __name__ == "__main__":
|
| 145 |
-
demo.launch(share=True)
|
|
|
|
| 1 |
+
python
|
| 2 |
+
|
| 3 |
import gradio as gr
|
| 4 |
+
from openai import OpenAI
|
| 5 |
import requests
|
| 6 |
import json
|
| 7 |
import os
|
|
|
|
| 20 |
def __init__(self):
|
| 21 |
self.openai_api_key = OPENAI_API_KEY
|
| 22 |
self.raindrop_api_token = RAINDROP_TOKEN
|
| 23 |
+
self.client = OpenAI(api_key=self.openai_api_key)
|
| 24 |
|
| 25 |
def generate_search_query(self, user_request: str) -> str:
|
| 26 |
"""Convert user request to a tailored search query using OpenAI."""
|
|
|
|
| 33 |
Format the search query with relevant keywords and operators if needed.
|
| 34 |
"""
|
| 35 |
|
| 36 |
+
response = self.client.chat.completions.create(
|
| 37 |
+
model="gpt-4o-mini",
|
| 38 |
+
messages=[{"role": "user", "content": prompt}],
|
| 39 |
+
temperature=0.5,
|
| 40 |
+
max_tokens=1000
|
|
|
|
| 41 |
)
|
| 42 |
+
return response.choices[0].message.content
|
|
|
|
| 43 |
|
| 44 |
def search_raindrop(self, search_query: str) -> List[Dict]:
|
| 45 |
"""Search Raindrop.io with the generated query."""
|
|
|
|
| 122 |
with gr.Row():
|
| 123 |
output_text = gr.Textbox(
|
| 124 |
label="Search Results",
|
| 125 |
+
lines=10,
|
| 126 |
+
readonly=True
|
| 127 |
)
|
| 128 |
|
| 129 |
search_button.click(
|
|
|
|
| 143 |
|
| 144 |
# Launch the interface
|
| 145 |
if __name__ == "__main__":
|
| 146 |
+
demo.launch(share=True)
|