Spaces:
Sleeping
Sleeping
| import GPTDriveIntegration | |
| import gradio as gr | |
| gpt_drive = GPTDriveIntegration() | |
| def process_user_query(query, search_terms_input): | |
| """Process user query and return formatted response""" | |
| if not query.strip(): | |
| return "Please enter a question.", "" | |
| # Parse search terms if provided | |
| search_terms = None | |
| # if search_terms_input.strip(): | |
| # search_terms = [term.strip() for term in search_terms_input.split(',')] | |
| # Process the query | |
| result = gpt_drive.process_query(query, search_terms) | |
| # Format the response | |
| answer = result['answer'] | |
| sources = result['sources'] | |
| sources_text = "" | |
| if sources: | |
| sources_text = "**Sources used:**\n" + "\n".join([f"• {source}" for source in sources]) | |
| return answer, sources_text | |
| def check_setup(): | |
| """Check if the APIs are properly configured""" | |
| status_messages = [] | |
| # Check Google Drive API | |
| if gpt_drive.drive_initialized: | |
| status_messages.append("✅ Google Drive API: Connected") | |
| else: | |
| status_messages.append(f"❌ Google Drive API: {getattr(gpt_drive, 'drive_error', 'Not configured')}") | |
| # Check OpenAI API | |
| if gpt_drive.openai_initialized: | |
| status_messages.append("✅ OpenAI API: Connected") | |
| else: | |
| status_messages.append(f"❌ OpenAI API: {getattr(gpt_drive, 'openai_error', 'Not configured')}") | |
| return "\n".join(status_messages) | |
| # Create Gradio interface | |
| with gr.Blocks(title="Augusta's Anatomy Reading Assistant", theme=gr.themes.Soft()) as app: | |
| gr.Markdown("# 🤖 Augusta's Anatomy bot") | |
| gr.Markdown("Ask questions about your anatomy books using AI!") | |
| with gr.Row(): | |
| with gr.Column(scale=2): | |
| # Main query interface | |
| with gr.Group(): | |
| gr.Markdown("### Ask a Question") | |
| query_input = gr.Textbox( | |
| label="Your Question", | |
| placeholder="Ask me any question about your anatomy books?", | |
| lines=3 | |
| ) | |
| search_terms_input = gr.Textbox( | |
| label="Search Terms (optional)", | |
| placeholder="Enter comma-separated terms to search for specific files", | |
| lines=1 | |
| ) | |
| submit_btn = gr.Button("Search & Ask", variant="primary", size="lg") | |
| # Results section | |
| with gr.Group(): | |
| gr.Markdown("### Answer") | |
| answer_output = gr.Textbox( | |
| label="AI Response", | |
| lines=10, | |
| interactive=False | |
| ) | |
| sources_output = gr.Textbox( | |
| label="Sources", | |
| lines=3, | |
| interactive=False | |
| ) | |
| with gr.Column(scale=1): | |
| # Status and setup info | |
| with gr.Group(): | |
| gr.Markdown("### System Status") | |
| status_btn = gr.Button("Check Status", size="sm") | |
| status_output = gr.Textbox( | |
| label="API Status", | |
| lines=4, | |
| interactive=False | |
| ) | |
| with gr.Group(): | |
| gr.Markdown("### Setup Instructions") | |
| gr.Markdown(""" | |
| **Important Notes:** | |
| 1.Only documents shared with it, it can answer | |
| **File Types Supported:** | |
| - Google Docs | |
| - Google Sheets | |
| - PDF files | |
| - Text files | |
| **Tips:** | |
| - Use specific search terms for better results | |
| - The system searches the top 3 most relevant files | |
| - Ask clear, specific questions for better answers | |
| """) | |
| # Event handlers | |
| submit_btn.click( | |
| fn=process_user_query, | |
| inputs=[query_input, search_terms_input], | |
| outputs=[answer_output, sources_output] | |
| ) | |
| status_btn.click( | |
| fn=check_setup, | |
| outputs=status_output | |
| ) | |
| # Example queries | |
| with gr.Row(): | |
| gr.Examples( | |
| examples=[ | |
| ["What is morbid Anatomy?", "morbid, Anatomy"], | |
| ["The transmission of nerves from one neuron to another is as a result of what?", "neuron, nerves, Dr Clement"], | |
| ], | |
| inputs=[query_input, search_terms_input], | |
| ) | |
| # Launch the app | |
| if __name__ == "__main__": | |
| app.launch( | |
| share=True,debug =True) |