Spaces:
Sleeping
Sleeping
| import gradio as gr | |
| import requests | |
| import os | |
| import re | |
| # WARNING: It is not recommended to hardcode sensitive data like API tokens in code. | |
| # Consider using environment variables or other secure methods for production applications. | |
| API_URL = "https://deployment.datasaur.ai/api/deployment/8/2723/chat/completions" | |
| API_TOKEN = os.environ["DATASAUR_API_KEY"] | |
| import re | |
| import json | |
| def extract_suggestion_value(text): | |
| """ | |
| Extracts the value of the 'suggestion' key from a given string that may contain JSON-like content. | |
| Parameters: | |
| text (str): Input text containing a JSON-like 'suggestion' field. | |
| Returns: | |
| dict: Dictionary with the 'suggestion' key and its extracted value, or None if not found. | |
| """ | |
| # First, try to directly parse as JSON | |
| try: | |
| parsed = json.loads(text) | |
| if "suggestion" in parsed: | |
| return {"suggestion": parsed["suggestion"]} | |
| except json.JSONDecodeError: | |
| pass # Fall back to regex if not valid JSON | |
| # Fallback: use regex to extract suggestion value | |
| match = re.search(r'"suggestion"\s*:\s*"([^"]+)"', text) | |
| if match: | |
| return {"suggestion": match.group(1)} | |
| return None | |
| def magic_function(input_text): | |
| """ | |
| Sends text to the Datasaur deployment API and returns the processed text. | |
| """ | |
| headers = { | |
| "Content-Type": "application/json", | |
| "Authorization": f"Bearer {API_TOKEN}", | |
| } | |
| data = { | |
| "messages": [{"role": "user", "content": f"Input: `{input_text}`"}] | |
| } | |
| try: | |
| response = requests.post(API_URL, headers=headers, json=data) | |
| response.raise_for_status() # Raise an exception for bad status codes (4xx or 5xx) | |
| response_json = response.json() | |
| # Extract content from a standard chat completion response structure. | |
| # This may need adjustment if the API has a different format. | |
| content = response_json.get("choices", [{}])[0].get("message", {}).get("content", "Error: Could not parse response.") | |
| print(content) | |
| content = extract_suggestion_value(content) | |
| print("----") | |
| print(content) | |
| return content | |
| except requests.exceptions.RequestException as e: | |
| return f"API Request Error: {e}" | |
| except (ValueError, KeyError, IndexError): | |
| # Handle cases where response is not valid JSON or structure is unexpected | |
| return f"Error processing API response: {response.text}" | |
| def handle_magic_click(current_text): | |
| """ | |
| When the magic button is clicked, this function gets the improved text, | |
| and returns the new and previous text to update the UI. | |
| It also makes the "Previous Text" box visible. | |
| """ | |
| improved_text = magic_function(current_text) | |
| return improved_text, current_text, gr.update(visible=True) | |
| with gr.Blocks() as demo: | |
| gr.Markdown("# Time Entry Improvement") | |
| with gr.Row(): | |
| with gr.Column(scale=4): | |
| text_area = gr.Textbox(label="Your Text", lines=5) | |
| previous_text_area = gr.Textbox(label="Previous Text", lines=5, visible=False) | |
| with gr.Column(scale=1): | |
| magic_button = gr.Button("Magic Button") | |
| magic_button.click( | |
| fn=handle_magic_click, | |
| inputs=text_area, | |
| outputs=[text_area, previous_text_area, previous_text_area] | |
| ) | |
| if __name__ == "__main__": | |
| demo.launch() | |