Create app.py

app.py

@@ -0,0 +1,146 @@
+import gradio as gr
+import openai
+import requests
+import json
+import os
+from typing import Dict, List
+
+# Get API keys from environment variables
+OPENAI_API_KEY = os.getenv('openaikey')
+RAINDROP_TOKEN = os.getenv('raindroptoken')
+
+if not OPENAI_API_KEY or not RAINDROP_TOKEN:
+    raise EnvironmentError(
+        "Missing required environment variables. Please ensure 'openaikey' and 'raindroptoken' are set."
+    )
+
+class RaindropSearchBot:
+    def __init__(self):
+        self.openai_api_key = OPENAI_API_KEY
+        self.raindrop_api_token = RAINDROP_TOKEN
+        openai.api_key = self.openai_api_key
+        
+    def generate_search_query(self, user_request: str) -> str:
+        """Convert user request to a tailored search query using OpenAI."""
+        prompt = f"""
+        Convert the following request into a focused search query for Raindrop.io.
+        The query should be specific and relevant for searching bookmarked links.
+        
+        User Request: {user_request}
+        
+        Format the search query with relevant keywords and operators if needed.
+        """
+        
+        response = openai.ChatCompletion.create(
+            model="gpt-3.5-turbo",
+            messages=[
+                {"role": "system", "content": "You are a search query optimization assistant."},
+                {"role": "user", "content": prompt}
+            ]
+        )
+        
+        return response.choices[0].message.content.strip()
+    
+    def search_raindrop(self, search_query: str) -> List[Dict]:
+        """Search Raindrop.io with the generated query."""
+        headers = {
+            "Authorization": f"Bearer {self.raindrop_api_token}",
+            "Content-Type": "application/json"
+        }
+        
+        params = {
+            "search": search_query,
+            "page": 0,
+            "perpage": 10
+        }
+        
+        response = requests.get(
+            "https://api.raindrop.io/rest/v1/raindrops/0",
+            headers=headers,
+            params=params
+        )
+        
+        if response.status_code == 200:
+            return response.json().get("items", [])
+        else:
+            return []
+    
+    def format_results(self, results: List[Dict]) -> str:
+        """Format the search results into a readable string."""
+        if not results:
+            return "No results found."
+            
+        formatted_output = "🔍 Search Results:\n\n"
+        for idx, item in enumerate(results, 1):
+            formatted_output += f"{idx}. {item.get('title', 'No Title')}\n"
+            formatted_output += f"   Link: {item.get('link', 'No Link')}\n"
+            if item.get('tags'):
+                formatted_output += f"   Tags: {', '.join(item['tags'])}\n"
+            formatted_output += "\n"
+            
+        return formatted_output
+    
+    def process_request(self, user_request: str) -> str:
+        """Process the user request and return formatted results."""
+        try:
+            # Generate optimized search query
+            search_query = self.generate_search_query(user_request)
+            
+            # Search Raindrop.io
+            results = self.search_raindrop(search_query)
+            
+            # Format and return results
+            return self.format_results(results)
+            
+        except Exception as e:
+            return f"An error occurred: {str(e)}"
+
+# Initialize the bot
+bot = RaindropSearchBot()
+
+# Create Gradio interface
+def chatbot_interface(user_input: str) -> str:
+    return bot.process_request(user_input)
+
+# Define and launch the Gradio interface
+with gr.Blocks(title="Raindrop.io Link Search Assistant", theme=gr.themes.Soft()) as demo:
+    gr.Markdown("""
+    # 🔍 Raindrop.io Link Search Assistant
+    Enter your search request in natural language, and I'll find relevant bookmarked links from your Raindrop.io collection.
+    """)
+    
+    with gr.Row():
+        input_text = gr.Textbox(
+            label="What would you like to search for?",
+            placeholder="Ex: Give me the list of all the links related to blockchain in Hong Kong",
+            lines=2
+        )
+    
+    with gr.Row():
+        search_button = gr.Button("🔍 Search", variant="primary")
+    
+    with gr.Row():
+        output_text = gr.Textbox(
+            label="Search Results",
+            lines=10,
+            readonly=True
+        )
+    
+    search_button.click(
+        fn=chatbot_interface,
+        inputs=input_text,
+        outputs=output_text
+    )
+    
+    gr.Markdown("""
+    ### How to use:
+    1. Enter your search request in natural language
+    2. Click the Search button
+    3. View the results from your Raindrop.io bookmarks
+    
+    The assistant will convert your request into an optimized search query and fetch relevant results.
+    """)
+
+# Launch the interface
+if __name__ == "__main__":
+    demo.launch(share=True)