Upload 4 files

app.py

@@ -0,0 +1,513 @@
+from flask import Flask, render_template, request, jsonify, send_file
+import ollama
+import json
+import re
+from duckduckgo_search import DDGS
+import requests
+from bs4 import BeautifulSoup
+import fitz  # PyMuPDF
+import urllib3
+import pandas as pd
+import io
+import ast
+
+
+app = Flask(__name__)
+
+search_prompt = """
+The user will provide a detailed description of a technical problem they are trying to solve in the context of intellectual property (IP) and patents. Your task is to generate some (2 to 5) highly specific and relevant search queries for Google, aimed at finding research papers closely related to the user's problem. Each search query should:
+
+1. Be crafted to find research papers, articles, or academic resources that address similar issues or solutions.
+2. Be focused and precise, avoiding generic or overly broad terms.
+
+Provide the search queries in the following **JSON format**. There should be no extra text, only the search queries as values.
+
+**Example Output:**
+
+```json
+{
+    "1": "user authentication 5G cryptographic keys identity management",
+    "2": "5G authentication security issues cryptography 3GPP key management"
+}
+```
+"""
+
+infringement_prompt = """You are an expert assistant designed to evaluate the novelty and inventiveness of patents by comparing them with existing documents. Your task is to analyze the background of a given patent and the first page of a related document to determine how well the document covers the problems mentioned in the patent.
+
+# Instructions:
+
+Understand the Patent Background: Carefully read and comprehend the background information provided for the patent. Identify the key problems that the patent aims to address.
+
+Analyze the Document: Review the provided document. Focus on identifying any problems that are similar to those mentioned in the patent background.
+
+Evaluate Coverage: Assess how well the document covers the problems mentioned in the patent. Use the following scoring system:
+
+Score 5: The document explicitly discusses the same problems as the patent, indicating that the problems are not novel.
+Score 4: The document discusses problems that are very similar to those in the patent, significantly impacting the novelty of the patent's problems.
+Score 3: The document mentions problems that are somewhat similar to those in the patent, but the coverage is not extensive enough to fully block the novelty of the patent's problems.
+Score 2: The document mentions problems that are similar in some ways but are clearly different from those in the patent.
+Score 1: The document touches upon related problems but does not directly address the specific problems mentioned in the patent.
+Score 0: The document does not discuss any problems related to those in the patent.
+Provide a Score: Based on your analysis, provide a score from 0 to 5 indicating how well the document covers the problems mentioned in the patent.
+
+Justify Your Score: Briefly explain the reasoning behind your score, highlighting specific similarities or differences between the problems discussed in the patent and the document.
+
+# Output Format:
+No details or explanations are required, just the results in the required **JSON** format with no additional word.
+
+{
+    'score': [Your Score],
+    'justification': "[Your Justification]"
+}
+"""
+
+insight_prompt = """Analyze the technical document and extract key insights that could enhance the patent problem. Focus on identifying security vulnerabilities, technical problems, innovative technologies, research questions, and protocols mentioned in the document.
+
+# Instructions:
+1. Identify 3-5 key insights from the document that could enhance or inform the patent problem.
+2. Each insight should be concise (max 10 words) but informative.
+3. Focus on technical elements that could be valuable for improving the patent.
+
+# Output Format:
+Return only a JSON object with insights as an array, with no additional text:
+
+{
+    "insights": [
+        "Security vulnerability in authentication handshake",
+        "Novel quantum encryption protocol",
+        "Lightweight implementation for IoT devices",
+        "Privacy-preserving key exchange mechanism",
+        "Real-time threat detection algorithm"
+    ]
+}
+"""
+
+refine_problem_prompt = """You are an expert consultant specializing in enhancing and refining technical problems to make them more sophisticated, novel, and inventive. Your task is to transform an initial technical problem description into two improved versions by integrating selected insights from technical documents.
+
+# Instructions:
+1. Carefully review the initial technical problem.
+2. Consider the selected insights from technical documents that could enhance the problem formulation.
+3. If any user comments are provided, incorporate those suggestions into your refinement process.
+4. Generate TWO distinct refined problem statements that:
+   - Integrate the selected insights in a meaningful way
+   - Make the problem more sophisticated and technically interesting
+   - Increase the novelty and inventive potential of any solution
+   - Add appropriate constraints and technical requirements
+   - Maintain coherence and technical feasibility
+   - Suggest innovative directions without fully solving the problem
+
+# Output Format:
+Return the results in JSON format with no additional text:
+
+{
+    "refined_problem_1": {
+        "title": "Brief descriptive title for the first refined problem",
+        "description": "Comprehensive description of the first refined problem that integrates insights and adds sophistication"
+    },
+    "refined_problem_2": {
+        "title": "Brief descriptive title for the second refined problem",
+        "description": "Alternative comprehensive description that takes a different approach to refining the problem"
+    }
+}
+"""
+
+
+def ask_ollama(user_message, model='gemma3:1b', system_prompt=search_prompt):
+    response = ollama.chat(model=model, messages=[
+        {
+            "role": "system",
+            "content": system_prompt
+        },
+        {
+            "role": "user", 
+            "content": user_message
+        }
+    ])
+    ai_reply = response['message']['content']
+    print(f"AI REPLY json:\n{ai_reply}")
+
+    # Process the response to ensure we return valid JSON
+    try:
+        # First, try to parse it directly in case it's already valid JSON
+        print(f"AI REPLY:\n{ai_reply}")
+        return ast.literal_eval(ai_reply.replace('json\n', '').replace('```', ''))
+    except:
+        print(f"ERROR:\n{e}")
+        # If it's not valid JSON, try to extract JSON from the text
+        return {
+            "1": "Error parsing response. Please try again.",
+            "2": "Error parsing response. Please try again."
+        }
+
+def search_web(topic, max_references=5, data_type="pdf"):
+    """Search the web using DuckDuckGo and return results."""
+    doc_list = []
+    with DDGS(verify=False) as ddgs:
+        i = 0
+        for r in ddgs.text(topic, region='wt-wt', safesearch='On', timelimit='n'):
+            if i >= max_references:
+                break
+            doc_list.append({"type": data_type, "title": r['title'], "body": r['body'], "url": r['href']})
+            i += 1
+    return doc_list
+
+def analyze_pdf_novelty(patent_background, url, data_type="pdf"):
+    """Extract full document text from PDF or background from patent and evaluate novelty"""
+    try:
+        # Disable SSL warnings
+        urllib3.disable_warnings(urllib3.exceptions.InsecureRequestWarning)
+        # Extract text based on the type
+        if data_type == "pdf":            
+            headers = {
+                "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.124 Safari/537.36",
+                "Accept": "application/pdf"
+            }
+
+            response = requests.get(url, headers=headers, timeout=20, verify=False)
+            if response.status_code != 200:
+                print(f"Failed to download PDF (status code: {response.status_code})")
+                return {"error": f"Failed to download PDF (status code: {response.status_code})"}
+                
+            # Extract full document text (limited to 6000 characters)
+            try:
+                pdf_document = fitz.open(stream=response.content, filetype="pdf")
+                if pdf_document.page_count == 0:
+                    return {"error": "PDF has no pages"}
+                    
+                text = ""
+                for page_num in range(min(5, pdf_document.page_count)):  # Limit to first 5 pages
+                    page = pdf_document.load_page(page_num)
+                    text += page.get_text() + " "
+                    if len(text) >= 6000:
+                        break
+                
+                text = text[:6000]  # Limit to 6000 characters
+            except Exception as e:
+                return {"error": f"Error processing PDF: {str(e)}"}
+        
+        elif data_type == "patent":
+            # Extract background from patent
+            print("extract from patent")
+            try:
+                response = requests.get(url, timeout=20, verify=False)
+                if response.status_code != 200:
+                    print(f"Failed to access patent (status code: {response.status_code})")
+                    return {"error": f"Failed to access patent (status code: {response.status_code})"}
+                content = response.content.decode('utf-8').replace("\n", "")
+                soup = BeautifulSoup(content, 'html.parser')
+                section = soup.find('section', itemprop='description', itemscope='')
+                matches = re.findall(r"background(.*?)(?:summary|description of the drawing)", str(section), re.DOTALL | re.IGNORECASE)
+                if matches:
+                    text = BeautifulSoup(matches[0], "html.parser").get_text(separator=" ").strip()
+                else:
+                    text = "Background section not found in patent."
+            except Exception as e:
+                return {"error": f"Error processing patent: {str(e)}"}
+        elif data_type == "web":  
+            try:
+                headers = {
+                    "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.124 Safari/537.36",
+                    "Accept": "application/pdf"
+                }
+                response = requests.get(url, headers=headers, timeout=20, verify=False)
+                response.raise_for_status()                
+                soup = BeautifulSoup(response.text, 'html.parser')
+                full_text = soup.get_text()
+                text = re.sub(r'\n+', ' ', full_text)[:5000]
+            except requests.RequestException as e:
+                return {"error": f"Error fetching the page: {str(e)}"}
+        else:
+            return {"error": "Unknown document type"}
+        
+        # Analyze with Ollama for novelty assessment
+        result = ask_ollama(
+            user_message=f"Patent background:\n{patent_background}\n\nDocument content:\n{text}",
+            system_prompt=infringement_prompt
+        )
+        
+        # # Extract insights
+        # insights = ask_ollama(
+        #     user_message=f"Document content:\n{text}",
+        #     system_prompt=insight_prompt
+        # )
+        
+        # # Combine results
+        # result['insights'] = insights.get('insights', [])
+        
+        return result
+        
+    except Exception as e:
+        return {"error": f"Error: {str(e)}"}
+
+
+def extract_insights(patent_background, url, data_type="pdf"):
+    """Extract full document text from PDF or background from patent and evaluate novelty"""
+    try:
+        # Disable SSL warnings
+        urllib3.disable_warnings(urllib3.exceptions.InsecureRequestWarning)
+        # Extract text based on the type
+        if data_type == "pdf":            
+            headers = {
+                "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.124 Safari/537.36",
+                "Accept": "application/pdf"
+            }
+
+            response = requests.get(url, headers=headers, timeout=20, verify=False)
+            if response.status_code != 200:
+                print(f"Failed to download PDF (status code: {response.status_code})")
+                return {"error": f"Failed to download PDF (status code: {response.status_code})"}
+                
+            # Extract full document text (limited to 6000 characters)
+            try:
+                pdf_document = fitz.open(stream=response.content, filetype="pdf")
+                if pdf_document.page_count == 0:
+                    return {"error": "PDF has no pages"}
+                    
+                text = ""
+                for page_num in range(min(5, pdf_document.page_count)):  # Limit to first 5 pages
+                    page = pdf_document.load_page(page_num)
+                    text += page.get_text() + " "
+                    if len(text) >= 6000:
+                        break
+                
+                text = text[:6000]  # Limit to 6000 characters
+            except Exception as e:
+                return {"error": f"Error processing PDF: {str(e)}"}
+        
+        elif data_type == "patent":
+            # Extract background from patent
+            print("extract from patent")
+            try:
+                response = requests.get(url, timeout=20, verify=False)
+                if response.status_code != 200:
+                    print(f"Failed to access patent (status code: {response.status_code})")
+                    return {"error": f"Failed to access patent (status code: {response.status_code})"}
+                content = response.content.decode('utf-8').replace("\n", "")
+                soup = BeautifulSoup(content, 'html.parser')
+                section = soup.find('section', itemprop='description', itemscope='')
+                matches = re.findall(r"background(.*?)(?:summary|description of the drawing)", str(section), re.DOTALL | re.IGNORECASE)
+                if matches:
+                    text = BeautifulSoup(matches[0], "html.parser").get_text(separator=" ").strip()
+                else:
+                    text = "Background section not found in patent."
+            except Exception as e:
+                return {"error": f"Error processing patent: {str(e)}"}
+        elif data_type == "web":  
+            try:
+                headers = {
+                    "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.124 Safari/537.36",
+                    "Accept": "application/pdf"
+                }
+                response = requests.get(url, headers=headers, timeout=20, verify=False)
+                response.raise_for_status()                
+                soup = BeautifulSoup(response.text, 'html.parser')
+                full_text = soup.get_text()
+                text = re.sub(r'\n+', ' ', full_text)[:5000]
+            except requests.RequestException as e:
+                return {"error": f"Error fetching the page: {str(e)}"}
+        else:
+            return {"error": "Unknown document type"}
+        
+        # Extract insights
+        insights = ask_ollama(
+            user_message=f"Document content:\n{text}",
+            system_prompt=insight_prompt
+        )
+        
+        return insights
+        
+    except Exception as e:
+        return {"error": f"Error: {str(e)}"}
+
+@app.route('/')
+def home():
+    return render_template('index.html')
+
+@app.route('/chat', methods=['POST'])
+def chat():
+    user_message = request.form.get('message')
+    ai_reply = ask_ollama(user_message)
+    return jsonify({'reply': ai_reply})
+
+@app.route('/search', methods=['POST'])
+def search():
+    query = request.form.get('query')
+    pdf_checked = request.form.get('pdfOption') == 'true'
+    patent_checked = request.form.get('patentOption') == 'true'
+    web_checked = request.form.get('webOption') == 'true' or request.form.get('webOption') == 'on'
+
+    if not query:
+        return jsonify({'error': 'No query provided', 'results': []})
+    
+    all_results = []
+    
+    try:
+        # Handle various combinations
+        if pdf_checked:
+            pdf_query = f"{query} filetype:pdf"
+            pdf_results = search_web(pdf_query, max_references=5, data_type="pdf")
+            all_results.extend(pdf_results)
+            
+        if patent_checked:
+            patent_query = f"{query} site:patents.google.com"
+            patent_results = search_web(patent_query, max_references=5, data_type="patent")
+            all_results.extend(patent_results)
+            
+        if web_checked:
+            # For web, we don't add anything to the query
+            web_results = search_web(query, max_references=5, data_type="web")
+            all_results.extend(web_results)
+            
+        # If nothing is checked, default to web search
+        if not (pdf_checked or patent_checked or web_checked):
+            web_results = search_web(query, max_references=5, data_type="web")
+            all_results.extend(web_results)
+        
+        return jsonify({'results': all_results})
+    except Exception as e:
+        print(f"Error performing search: {e}")
+        return jsonify({'error': str(e), 'results': []})
+
+@app.route('/analyze', methods=['POST'])
+def analyze():
+    data = request.json
+    if not data or 'patent_background' not in data or 'pdf_url' not in data:
+        return jsonify({'error': 'Missing required parameters', 'result': None})
+    
+    try:
+        patent_background = data['patent_background']
+        url = data['pdf_url']
+        data_type = data.get('data_type', 'pdf')  # Default to pdf if not specified
+        
+        result = analyze_pdf_novelty(patent_background, url, data_type)
+        return jsonify({'result': result})
+    except Exception as e:
+        print(f"Error analyzing document: {e}")
+        return jsonify({'error': str(e), 'result': None})
+   
+@app.route('/post_insights', methods=['POST'])
+def post_insights():
+    data = request.json
+    if not data or 'patent_background' not in data or 'pdf_url' not in data:
+        return jsonify({'error': 'Missing required parameters', 'result': None})
+    
+    try:
+        patent_background = data['patent_background']
+        url = data['pdf_url']
+        data_type = data.get('data_type', 'pdf')  # Default to pdf if not specified
+        
+        result = extract_insights(patent_background, url, data_type)
+        return jsonify({'result': result})
+    except Exception as e:
+        print(f"Error analyzing document: {e}")
+        return jsonify({'error': str(e), 'result': None})
+
+@app.route('/refine-problem', methods=['POST'])
+def refine_problem():
+    data = request.json
+    if not data or 'original_problem' not in data or 'insights' not in data:
+        return jsonify({'error': 'Missing required parameters', 'result': None})
+    
+    try:
+        original_problem = data['original_problem']
+        insights = data['insights']
+        user_comments = data.get('user_comments', '')
+        
+        # Prepare the message for the LLM
+        message = f"""Initial Technical Problem:
+{original_problem}
+
+Selected Insights:
+{', '.join(insights)}
+
+User Comments/Suggestions:
+{user_comments}
+"""
+        
+        # Get refined problem suggestions from the LLM
+        result = ask_ollama(
+            user_message=message,
+            system_prompt=refine_problem_prompt
+        )
+        
+        return jsonify({'result': result})
+    except Exception as e:
+        print(f"Error refining problem: {e}")
+        return jsonify({'error': str(e), 'result': None})
+    
+@app.route('/export-excel', methods=['POST'])
+def export_excel():
+    try:
+        data = request.json
+        if not data or 'tableData' not in data:
+            return jsonify({'error': 'No table data provided'})
+            
+        # Create pandas DataFrame from the data
+        df = pd.DataFrame(data['tableData'])
+        
+        # Get the user query
+        user_query = data.get('userQuery', 'No query provided')
+        
+        # Get problem versions if available
+        problem_versions = data.get('problemVersions', {})
+        
+        # Create a BytesIO object to store the Excel file
+        output = io.BytesIO()
+        
+        # Create Excel file with xlsxwriter engine
+        with pd.ExcelWriter(output, engine='xlsxwriter') as writer:
+            # Write the data to a sheet named 'Results'
+            df.to_excel(writer, sheet_name='Results', index=False)
+            
+            # Get workbook and worksheet objects
+            workbook = writer.book
+            worksheet = writer.sheets['Results']
+            
+            # Add a sheet for the query
+            query_sheet = workbook.add_worksheet('Query')
+            query_sheet.write(0, 0, 'Patent Query')
+            query_sheet.write(1, 0, user_query)
+            
+            # Add a sheet for problem versions if available
+            if problem_versions:
+                versions_sheet = workbook.add_worksheet('Problem Versions')
+                versions_sheet.write(0, 0, 'Version')
+                versions_sheet.write(0, 1, 'Problem Statement')
+                
+                row = 1
+                for version, text in problem_versions.items():
+                    versions_sheet.write(row, 0, version)
+                    versions_sheet.write(row, 1, text)
+                    row += 1
+                
+                # Set column width for problem versions
+                versions_sheet.set_column(0, 0, 20)
+                versions_sheet.set_column(1, 1, 100)
+            
+            # Adjust column widths
+            for i, col in enumerate(df.columns):
+                # Get maximum column width
+                max_len = max(
+                    df[col].astype(str).map(len).max(),
+                    len(col)
+                ) + 2
+                # Set column width (limit to 100 to avoid issues)
+                worksheet.set_column(i, i, min(max_len, 100))
+        
+        # Seek to the beginning of the BytesIO object
+        output.seek(0)
+        
+        # Return the Excel file
+        return send_file(
+            output,
+            mimetype='application/vnd.openxmlformats-officedocument.spreadsheetml.sheet',
+            as_attachment=True,
+            download_name='patent_search_results.xlsx'
+        )
+        
+    except Exception as e:
+        print(f"Error exporting Excel: {e}")
+        return jsonify({'error': str(e)})
+
+if __name__ == '__main__':
+    app.run(debug=True)

requirements.txt

@@ -0,0 +1,35 @@
+annotated-types==0.7.0
+anyio==4.9.0
+beautifulsoup4==4.13.3
+blinker==1.9.0
+certifi==2025.1.31
+charset-normalizer==3.4.1
+click==8.1.8
+duckduckgo_search==7.5.2
+Flask==3.1.0
+h11==0.14.0
+httpcore==1.0.7
+httpx==0.28.1
+idna==3.10
+itsdangerous==2.2.0
+Jinja2==3.1.6
+lxml==5.3.1
+MarkupSafe==3.0.2
+numpy==2.2.4
+ollama==0.4.7
+pandas==2.2.3
+primp==0.14.0
+pydantic==2.10.6
+pydantic_core==2.27.2
+PyMuPDF==1.25.4
+python-dateutil==2.9.0.post0
+pytz==2025.1
+requests==2.32.3
+six==1.17.0
+sniffio==1.3.1
+soupsieve==2.6
+typing_extensions==4.12.2
+tzdata==2025.1
+urllib3==2.3.0
+Werkzeug==3.1.3
+XlsxWriter==3.2.2

static/style.css

@@ -0,0 +1,628 @@
+:root {
+    --primary: #2563eb;
+    --primary-light: #3b82f6;
+    --primary-dark: #1d4ed8;
+    --danger: #ef4444;
+    --success: #10b981;
+    --warning: #f59e0b;
+    --gray-100: #f3f4f6;
+    --gray-200: #e5e7eb;
+    --gray-300: #d1d5db;
+    --gray-400: #9ca3af;
+    --gray-500: #6b7280;
+    --gray-600: #4b5563;
+    --gray-700: #374151;
+    --gray-800: #1f2937;
+    --gray-900: #111827;
+    --transition: all 0.2s ease;
+    --shadow-sm: 0 1px 2px 0 rgba(0, 0, 0, 0.05);
+    --shadow: 0 1px 3px 0 rgba(0, 0, 0, 0.1), 0 1px 2px 0 rgba(0, 0, 0, 0.06);
+    --shadow-md: 0 4px 6px -1px rgba(0, 0, 0, 0.1), 0 2px 4px -1px rgba(0, 0, 0, 0.06);
+    --shadow-lg: 0 10px 15px -3px rgba(0, 0, 0, 0.1), 0 4px 6px -2px rgba(0, 0, 0, 0.05);
+}
+
+* {
+    box-sizing: border-box;
+    margin: 0;
+    padding: 0;
+}
+
+body {
+    font-family: 'Inter', -apple-system, BlinkMacSystemFont, 'Segoe UI', Roboto, Helvetica, Arial, sans-serif;
+    line-height: 1.6;
+    color: var(--gray-800);
+    background-color: #f8fafc;
+    max-width: 1200px;
+    margin: 0 auto;
+    padding: 2rem;
+}
+
+@media (max-width: 768px) {
+    body {
+        padding: 1rem;
+    }
+}
+
+.container {
+    max-width: 1000px;
+    margin: 0 auto;
+}
+
+.header {
+    margin-bottom: 2.5rem;
+    text-align: center;
+}
+
+.logo {
+    display: flex;
+    align-items: center;
+    justify-content: center;
+    margin-bottom: 1rem;
+}
+
+.logo-icon {
+    width: 40px;
+    height: 40px;
+    background-color: var(--primary);
+    border-radius: 8px;
+    color: white;
+    display: flex;
+    align-items: center;
+    justify-content: center;
+    font-weight: bold;
+    font-size: 1.5rem;
+    margin-right: 10px;
+}
+
+h1 {
+    font-size: 2rem;
+    font-weight: 700;
+    color: var(--gray-900);
+    margin-bottom: 0.5rem;
+}
+
+h2 {
+    font-size: 1.5rem;
+    font-weight: 600;
+    color: var(--gray-800);
+    margin-bottom: 1.5rem;
+}
+
+p {
+    color: var(--gray-600);
+    margin-bottom: 1rem;
+}
+
+.card {
+    background-color: white;
+    border-radius: 0.75rem;
+    box-shadow: var(--shadow);
+    padding: 2rem;
+    margin-bottom: 2rem;
+}
+
+.form-group {
+    margin-bottom: 1.5rem;
+}
+
+label {
+    display: block;
+    margin-bottom: 0.5rem;
+    font-weight: 500;
+    color: var(--gray-700);
+}
+
+#userInput {
+    width: 100%;
+    padding: 0.75rem;
+    border: 1px solid var(--gray-300);
+    border-radius: 0.5rem;
+    min-height: 150px;
+    font-family: inherit;
+    resize: vertical;
+    transition: var(--transition);
+}
+
+#userInput:focus {
+    outline: none;
+    border-color: var(--primary);
+    box-shadow: 0 0 0 3px rgba(37, 99, 235, 0.2);
+}
+
+.btn {
+    display: inline-flex;
+    align-items: center;
+    justify-content: center;
+    padding: 0.625rem 1.25rem;
+    border: none;
+    border-radius: 0.5rem;
+    font-weight: 500;
+    font-size: 0.875rem;
+    cursor: pointer;
+    transition: var(--transition);
+    gap: 0.5rem;
+}
+
+.btn-primary {
+    background-color: var(--primary);
+    color: white;
+}
+
+.btn-primary:hover {
+    background-color: var(--primary-dark);
+}
+
+.btn-secondary {
+    background-color: var(--gray-100);
+    color: var(--gray-700);
+}
+
+.btn-secondary:hover {
+    background-color: var(--gray-200);
+}
+
+.btn-danger {
+    background-color: var(--danger);
+    color: white;
+}
+
+.btn-danger:hover {
+    background-color: #dc2626;
+}
+
+.btn-success {
+    background-color: var(--success);
+    color: white;
+}
+
+.btn-success:hover {
+    background-color: #059669;
+}
+
+.btn-warning {
+    background-color: var(--warning);
+    color: white;
+}
+
+.btn-warning:hover {
+    background-color: #d97706;
+}
+
+.query-item {
+    background-color: white;
+    border-radius: 0.75rem;
+    box-shadow: var(--shadow);
+    padding: 1.5rem;
+    margin-bottom: 1.5rem;
+}
+
+.query-container {
+    display: flex;
+    margin-bottom: 1rem;
+    align-items: center;
+    gap: 0.5rem;
+}
+
+.query-field {
+    flex-grow: 1;
+    padding: 0.625rem 0.75rem;
+    border: 1px solid var(--gray-300);
+    border-radius: 0.5rem;
+    transition: var(--transition);
+}
+
+.query-field:focus {
+    outline: none;
+    border-color: var(--primary);
+    box-shadow: 0 0 0 3px rgba(37, 99, 235, 0.2);
+}
+
+.action-button {
+    background-color: hsl(192, 83%, 66%);
+    color: white;
+    padding: 8px 12px;
+    border: none;
+    border-radius: 4px;
+    cursor: pointer;
+    margin-right: 5px;
+}
+.delete-button {
+    background-color: #f44336;
+}
+.search-button {
+    background-color: #2196F3;
+}
+button:hover {
+    opacity: 0.9;
+}
+
+#resultsContainer {
+    margin-top: 20px;
+    display: none;
+}
+#queriesContainer {
+    margin-bottom: 15px;
+}
+#loadingIndicator {
+    display: none;
+    margin-top: 20px;
+    text-align: center;
+    color: #666;
+}
+.button-container {
+    margin-top: 15px;
+}
+.results-table {
+    width: 100%;
+    border-collapse: collapse;
+    margin-top: 10px;
+    display: none;
+}
+.results-table th, .results-table td {
+    border: 1px solid #ddd;
+    padding: 8px;
+    text-align: left;
+}
+.results-table th {
+    background-color: #f2f2f2;
+}
+.results-table tr:nth-child(even) {
+    background-color: #f9f9f9;
+}
+.results-table tr:hover {
+    background-color: #f1f1f1;
+}
+.search-loading {
+    display: none;
+    margin: 10px 0;
+    font-style: italic;
+    color: #666;
+}
+.url-link {
+    color: #0066cc;
+    text-decoration: none;
+    word-break: break-all;
+}
+.url-link:hover {
+    text-decoration: underline;
+}
+
+.analyze-button {
+    margin-top: 5px;
+    padding: 3px 8px;
+    font-size: 0.8em;
+}
+
+.loading-spinner {
+    border: 4px solid #f3f3f3;
+    border-top: 4px solid #3498db;
+    border-radius: 50%;
+    width: 20px;
+    height: 20px;
+    animation: spin 2s linear infinite;
+    margin: 0 auto;
+}
+
+@keyframes spin {
+    0% { transform: rotate(0deg); }
+    100% { transform: rotate(360deg); }
+}
+
+.score-cell, .justification-cell {
+    max-width: 150px;
+    overflow: hidden;
+    text-overflow: ellipsis;
+}
+
+.score-cell {
+    font-weight: bold;
+    text-align: center;
+}
+
+.floating-buttons {
+    position: fixed;
+    bottom: 2rem;
+    right: 2rem;
+    display: flex;
+    flex-direction: column;
+    gap: 0.75rem;
+    z-index: 100;
+}
+
+.floating-button {
+    padding: 0.75rem;
+    border-radius: 0.5rem;
+    box-shadow: var(--shadow-md);
+    transition: all 0.3s ease;
+    display: flex;
+    align-items: center;
+    justify-content: center;
+    gap: 0.5rem;
+    width: auto;
+    white-space: nowrap;
+}
+
+.floating-button:hover {
+    transform: translateY(-2px);
+    box-shadow: var(--shadow-lg);
+}
+
+.loading-overlay {
+    position: fixed;
+    top: 0;
+    left: 0;
+    width: 100%;
+    height: 100%;
+    background-color: rgba(0, 0, 0, 0.5);
+    display: none;
+    justify-content: center;
+    align-items: center;
+    z-index: 1000;
+}
+
+.loading-content {
+    background-color: white;
+    padding: 2rem;
+    border-radius: 0.75rem;
+    text-align: center;
+    box-shadow: var(--shadow-lg);
+    max-width: 80%;
+}
+
+.loading-content .loading-spinner {
+    width: 40px;
+    height: 40px;
+    border-width: 4px;
+    margin-bottom: 1rem;
+}
+
+.progress-text {
+    font-size: 1rem;
+    margin-top: 0.75rem;
+    color: var(--gray-700);
+}
+
+.button-container {
+    margin-top: 1.5rem;
+    display: flex;
+    justify-content: flex-start;
+    gap: 0.75rem;
+}
+
+/* Add icons for better UX */
+.icon {
+    display: inline-block;
+    width: 20px;
+    height: 20px;
+    background-size: contain;
+    background-repeat: no-repeat;
+    background-position: center;
+}
+
+.table-container {
+    margin-top: 1rem;
+    overflow-x: auto;
+    border-radius: 0.5rem;
+    border: 1px solid var(--gray-200);
+}
+
+/* Responsive adjustments */
+@media (max-width: 768px) {
+    .query-container {
+        flex-direction: column;
+        align-items: stretch;
+    }
+    
+    .query-container .btn {
+        margin-top: 0.5rem;
+    }
+    
+    .floating-buttons {
+        bottom: 1rem;
+        right: 1rem;
+    }
+}
+
+.search-options {
+    margin-bottom: 20px;
+    text-align: center;
+}
+
+.checkbox-group {
+    display: flex;
+    gap: 15px;
+    margin-top: 5px;
+    justify-content: center; /* This centers the checkboxes horizontally */
+}
+
+.checkbox-item {
+    display: flex;
+    align-items: center;
+    gap: 5px;
+}
+
+/* Adding styles for insights feature */
+.insights-container {
+    margin-top: 10px;
+    border-top: 1px dashed #ccc;
+    padding-top: 10px;
+}
+
+.insights-header {
+    display: flex;
+    justify-content: space-between;
+    align-items: center;
+    margin-bottom: 8px;
+}
+
+.insights-title {
+    font-weight: bold;
+    color: #333;
+}
+
+.insights-actions {
+    display: flex;
+    gap: 8px;
+}
+
+.insights-button {
+    font-size: 12px;
+    padding: 2px 6px;
+    background-color: #4CAF50;
+    color: white;
+    border: none;
+    border-radius: 3px;
+    cursor: pointer;
+}
+
+.insights-list {
+    display: flex;
+    flex-wrap: wrap;
+    gap: 8px;
+    margin-top: 8px;
+}
+
+.insight-tag {
+    background-color: #f1f1f1;
+    border: 1px solid #ddd;
+    border-radius: 15px;
+    padding: 4px 12px;
+    font-size: 12px;
+    cursor: pointer;
+    transition: all 0.2s ease;
+    user-select: none;
+}
+
+.insight-tag.selected {
+    background-color: #4CAF50;
+    color: white;
+    border-color: #4CAF50;
+}
+
+/* Styles for problem refinement functionality */
+.problem-history {
+    position: relative;
+    margin-bottom: 10px;
+}
+
+.problem-history-nav {
+    display: flex;
+    align-items: center;
+    margin-bottom: 5px;
+}
+
+.history-arrow {
+    padding: 2px 8px;
+    background-color: var(--gray-200);
+    border-radius: 3px;
+    cursor: pointer;
+    margin-right: 5px;
+    font-size: 14px;
+    color: var(--gray-700);
+}
+
+.history-arrow:hover {
+    background-color: var(--gray-300);
+}
+
+.history-arrow.disabled {
+    opacity: 0.5;
+    cursor: not-allowed;
+}
+
+.history-status {
+    font-size: 12px;
+    color: var(--gray-600);
+}
+
+.insight-comment-area {
+    margin-top: 10px;
+    width: 100%;
+}
+
+.insight-comment-textarea {
+    width: 100%;
+    padding: 8px;
+    border: 1px solid var(--gray-300);
+    border-radius: 4px;
+    font-size: 14px;
+    resize: vertical;
+    min-height: 60px;
+}
+
+.refined-problem-container {
+    margin-top: 20px;
+    border: 1px solid var(--gray-300);
+    border-radius: 8px;
+    overflow: hidden;
+}
+
+.refined-problem-tabs {
+    display: flex;
+    background-color: var(--gray-100);
+    border-bottom: 1px solid var(--gray-300);
+}
+
+.refined-problem-tab {
+    padding: 10px 15px;
+    cursor: pointer;
+    font-weight: 500;
+    border-right: 1px solid var(--gray-300);
+}
+
+.refined-problem-tab.active {
+    background-color: white;
+    border-bottom: 2px solid var(--primary);
+}
+
+.refined-problem-content {
+    padding: 15px;
+    background-color: white;
+}
+
+.refined-problem {
+    display: none;
+}
+
+.refined-problem.active {
+    display: block;
+}
+
+.refined-problem-title {
+    font-weight: 600;
+    margin-bottom: 10px;
+    color: var(--primary-dark);
+}
+
+.refined-problem-description {
+    line-height: 1.5;
+}
+
+.apply-problem-btn {
+    margin-top: 10px;
+    background-color: var(--primary);
+    color: white;
+    border: none;
+    border-radius: 4px;
+    padding: 5px 10px;
+    cursor: pointer;
+    font-size: 14px;
+}
+
+.apply-problem-btn:hover {
+    background-color: var(--primary-dark);
+}
+
+.enhance-problem-button {
+    background-color: var(--warning);
+    color: white;
+}
+
+.enhance-problem-button:hover {
+    background-color: #d97706;
+}
+
+.enhance-problem-button.disabled {
+    opacity: 0.6;
+    cursor: not-allowed;
+}

templates/index.html

@@ -0,0 +1,1252 @@
+<!DOCTYPE html>
+<html lang="en">
+<head>
+    <meta charset="UTF-8">
+    <meta name="viewport" content="width=device-width, initial-scale=1.0">
+    <title>Patentability</title>
+    <link rel="stylesheet" href="{{ url_for('static', filename='style.css') }}">
+    <script>
+        // Array to store problem history versions
+        const problemHistory = [];
+        let currentHistoryIndex = -1;
+        // Store all refined problem suggestions
+        let storedRefinedProblems = {};
+        
+        function generateQueries(event) {
+            event.preventDefault();
+            
+            const userInput = document.getElementById('userInput').value.trim();
+            if (!userInput) {
+                alert('Please enter a technical problem description');
+                return;
+            }
+            
+            // Show loading indicator
+            document.getElementById('loadingIndicator').style.display = 'block';
+            // Hide results if they were previously shown
+            document.getElementById('resultsContainer').style.display = 'none';
+            
+            // Send message to Flask backend
+            fetch('/chat', {
+                method: 'POST',
+                body: new URLSearchParams({ 'message': userInput }),
+                headers: { 'Content-Type': 'application/x-www-form-urlencoded' }
+            })
+            .then(response => response.json())
+            .then(data => {
+                // Hide loading indicator
+                document.getElementById('loadingIndicator').style.display = 'none';
+                
+                try {
+                    // Parse the JSON string from the LLM response
+                    let jsonData;
+                    // First check if data.reply is already an object
+                    if (typeof data.reply === 'object') {
+                        jsonData = data.reply;
+                    } else {
+                        // Try to extract JSON if it's a string possibly with markdown code blocks
+                        const jsonString = data.reply.replace(/```json|```/g, '').trim();
+                        jsonData = JSON.parse(jsonString);
+                    }
+                    
+                    // Clear existing queries
+                    const queriesContainer = document.getElementById('queriesContainer');
+                    queriesContainer.innerHTML = '';
+                    
+                    // Add each query from the response
+                    Object.keys(jsonData).forEach(key => {
+                        addQueryField(jsonData[key]);
+                    });
+                    
+                    // Show results container
+                    document.getElementById('resultsContainer').style.display = 'block';
+                } catch (error) {
+                    console.error('Error parsing JSON:', error);
+                    alert('There was an error processing the response. Please try again.');
+                }
+            })
+            .catch(error => {
+                document.getElementById('loadingIndicator').style.display = 'none';
+                console.error('Error:', error);
+                alert('There was an error communicating with the server. Please try again.');
+            });
+        }
+        
+        // Add a new query field with the given text (or empty if not provided)
+        function addQueryField(queryText = '') {
+            const queriesContainer = document.getElementById('queriesContainer');
+            const queryIndex = queriesContainer.children.length + 1;
+            
+            // Create main container for this query item
+            const queryItem = document.createElement('div');
+            queryItem.className = 'query-item';
+            
+            // Create container for query field and buttons
+            const queryContainer = document.createElement('div');
+            queryContainer.className = 'query-container';
+            
+            // Input field
+            const queryField = document.createElement('input');
+            queryField.type = 'text';
+            queryField.className = 'query-field';
+            queryField.value = queryText;
+            queryField.placeholder = `Search Query ${queryIndex}`;
+            
+            // Delete button
+            const deleteButton = document.createElement('button');
+            deleteButton.type = 'button';
+            deleteButton.className = 'action-button delete-button';
+            deleteButton.textContent = 'Delete';
+            deleteButton.onclick = function() {
+                queriesContainer.removeChild(queryItem);
+                // Update the placeholder numbers for remaining queries
+                updateQueryIndices();
+            };
+            // Search button
+            const searchButton = document.createElement('button');
+            searchButton.type = 'button';
+            searchButton.className = 'action-button search-button';
+            searchButton.textContent = 'Search';
+            searchButton.onclick = function() {
+                performSearch(queryField.value, queryItem);
+            };
+            
+            // Add elements to container
+            queryContainer.appendChild(queryField);
+            queryContainer.appendChild(searchButton);
+            queryContainer.appendChild(deleteButton);
+            
+            // Create loading indicator for this search
+            const searchLoading = document.createElement('div');
+            searchLoading.className = 'search-loading';
+            searchLoading.textContent = 'Searching... Please wait.';
+            
+            // Create table for results
+            const resultsTable = document.createElement('table');
+            resultsTable.className = 'results-table';
+            
+            // Add table header
+            const tableHeader = document.createElement('thead');
+            const headerRow = document.createElement('tr');
+
+            const typeHeader = document.createElement('th');
+            typeHeader.textContent = 'Type';
+
+            const titleHeader = document.createElement('th');
+            titleHeader.textContent = 'Title';
+
+            const bodyHeader = document.createElement('th');
+            bodyHeader.textContent = 'Description';
+
+            const urlHeader = document.createElement('th');
+            urlHeader.textContent = 'URL';
+
+            const scoreHeader = document.createElement('th');
+            scoreHeader.textContent = 'Score';
+
+            const justificationHeader = document.createElement('th');
+            justificationHeader.textContent = 'Justification';
+
+            headerRow.appendChild(typeHeader);
+            headerRow.appendChild(titleHeader);
+            headerRow.appendChild(bodyHeader);
+            headerRow.appendChild(urlHeader);
+            headerRow.appendChild(scoreHeader);
+            headerRow.appendChild(justificationHeader);
+            tableHeader.appendChild(headerRow);
+            
+            // Add table body
+            const tableBody = document.createElement('tbody');
+            
+            resultsTable.appendChild(tableHeader);
+            resultsTable.appendChild(tableBody);
+            
+            // Add all elements to the query item
+            queryItem.appendChild(queryContainer);
+            queryItem.appendChild(searchLoading);
+            queryItem.appendChild(resultsTable);
+            
+            // Add container to the queries list
+            queriesContainer.appendChild(queryItem);
+        }
+        
+        // Perform search and update the results table
+        function performSearch(query, queryItemElement) {
+            if (!query.trim()) {
+                alert('Please enter a search query');
+                return;
+            }
+            
+            const loadingElement = queryItemElement.querySelector('.search-loading');
+            const tableElement = queryItemElement.querySelector('.results-table');
+            const tableBody = tableElement.querySelector('tbody');
+            
+            // Show loading and hide previous results
+            loadingElement.style.display = 'block';
+            tableElement.style.display = 'none';
+            
+            // Clear previous results
+            tableBody.innerHTML = '';
+            
+            // Get checkbox values
+            const pdfChecked = document.getElementById('pdfOption').checked;
+            const patentChecked = document.getElementById('patentOption').checked;
+            const webChecked = document.getElementById('webOption').checked;
+
+            
+            // Create form data with query and checkbox values
+            const formData = new FormData();
+            formData.append('query', query);
+            formData.append('pdfOption', pdfChecked);
+            formData.append('patentOption', patentChecked);
+            formData.append('webOption', webChecked);
+
+            
+            // Send search request to backend
+            fetch('/search', {
+                method: 'POST',
+                body: formData
+            })
+            .then(response => response.json())
+            .then(data => {
+                // Hide loading indicator
+                loadingElement.style.display = 'none';
+                
+                if (data.results && data.results.length > 0) {
+                    // Populate table with results
+                    data.results.forEach(result => {
+                        const row = document.createElement('tr');
+
+                        const typeCell = document.createElement('td');
+                        typeCell.textContent = result.type;                        
+                        
+                        const titleCell = document.createElement('td');
+                        titleCell.textContent = result.title;
+                        
+                        const bodyCell = document.createElement('td');
+                        bodyCell.textContent = result.body;
+                        
+                        const urlCell = document.createElement('td');
+                        const urlLink = document.createElement('a');
+                        urlLink.href = result.url;
+                        urlLink.textContent = result.url;
+                        urlLink.className = 'url-link';
+                        urlLink.target = '_blank';
+                        urlCell.appendChild(urlLink);
+
+                        // Add "Analyze" button to the URL cell
+                        const analyzeButton = document.createElement('button');
+                        analyzeButton.textContent = 'Analyze';
+                        analyzeButton.className = 'action-button analyze-button';
+                        analyzeButton.onclick = function(e) {
+                            e.preventDefault();
+                            analyzePaper(result.url, queryItemElement);
+                        };
+                        urlCell.appendChild(document.createElement('br'));
+                        urlCell.appendChild(analyzeButton);
+                        
+                        row.appendChild(typeCell);
+                        row.appendChild(titleCell);
+                        row.appendChild(bodyCell);
+                        row.appendChild(urlCell);
+                        
+                        tableBody.appendChild(row);
+                    });
+                    
+                    // Show the table
+                    tableElement.style.display = 'table';
+                } else {
+                    // No results
+                    const row = document.createElement('tr');
+                    const cell = document.createElement('td');
+                    cell.colSpan = 4; // Updated to 4 since we now have 4 columns with the type column
+                    cell.textContent = 'No results found.';
+                    cell.style.textAlign = 'center';
+                    row.appendChild(cell);
+                    tableBody.appendChild(row);
+                    tableElement.style.display = 'table';
+                }
+            })
+            .catch(error => {
+                loadingElement.style.display = 'none';
+                console.error('Error:', error);
+                
+                // Show error in table
+                const row = document.createElement('tr');
+                const cell = document.createElement('td');
+                cell.colSpan = 4; // Updated to 4 since we now have 4 columns with the type column
+                cell.textContent = 'Error performing search. Please try again.';
+                cell.style.textAlign = 'center';
+                cell.style.color = 'red';
+                row.appendChild(cell);
+                tableBody.appendChild(row);
+                tableElement.style.display = 'table';
+            });
+        }
+        
+        // Extract insights from document
+        function extractInsights(paperUrl, row) {
+            const patentBackground = document.getElementById('userInput').value.trim();
+            if (!patentBackground) {
+                alert('Please provide a patent background in the input field');
+                return;
+            }
+            
+            const documentType = row.cells[0].textContent.toLowerCase();
+            
+            // Create or get the insight cells
+            let insightsContainer;
+            if (row.querySelector('.insights-container')) {
+                insightsContainer = row.querySelector('.insights-container');
+                insightsContainer.innerHTML = '<div class="loading-spinner"></div><div>Extracting insights...</div>';
+            } else {
+                // Create insights cell that spans across all columns
+                const insightsRow = document.createElement('tr');
+                insightsRow.className = 'insights-row';
+                
+                const insightsTd = document.createElement('td');
+                insightsTd.colSpan = row.cells.length;
+                
+                insightsContainer = document.createElement('div');
+                insightsContainer.className = 'insights-container';
+                insightsContainer.innerHTML = '<div class="loading-spinner"></div><div>Extracting insights...</div>';
+                
+                insightsTd.appendChild(insightsContainer);
+                insightsRow.appendChild(insightsTd);
+                
+                // Insert after the current row
+                row.parentNode.insertBefore(insightsRow, row.nextSibling);
+            }
+            
+            // Send request to extract insights
+            fetch('/post_insights', {
+                method: 'POST',
+                body: JSON.stringify({ 
+                    'patent_background': patentBackground, 
+                    'pdf_url': paperUrl,
+                    'data_type': documentType
+                }),
+                headers: { 'Content-Type': 'application/json' }
+            })
+            .then(response => response.json())
+            .then(data => {
+                if (data.error) {
+                    insightsContainer.innerHTML = `<div class="error">Error: ${data.error}</div>`;
+                } else if (data.result && data.result.insights) {
+                    // Create header with title and actions
+                    const insightsHeader = document.createElement('div');
+                    insightsHeader.className = 'insights-header';
+                    
+                    const insightsTitle = document.createElement('div');
+                    insightsTitle.className = 'insights-title';
+                    insightsTitle.textContent = 'Document Insights';
+                    
+                    const insightsActions = document.createElement('div');
+                    insightsActions.className = 'insights-actions';
+                    
+                    const selectAllBtn = document.createElement('button');
+                    selectAllBtn.className = 'insights-button';
+                    selectAllBtn.textContent = 'Select All';
+                    selectAllBtn.onclick = function() {
+                        const tags = insightsContainer.querySelectorAll('.insight-tag');
+                        tags.forEach(tag => tag.classList.add('selected'));
+                        row.dataset.selectedInsights = JSON.stringify(data.result.insights);
+                        row.dataset.unselectedInsights = JSON.stringify([]);
+                    };
+                    
+                    const clearAllBtn = document.createElement('button');
+                    clearAllBtn.className = 'insights-button';
+                    clearAllBtn.textContent = 'Clear All';
+                    clearAllBtn.onclick = function() {
+                        const tags = insightsContainer.querySelectorAll('.insight-tag');
+                        tags.forEach(tag => tag.classList.remove('selected'));
+                        row.dataset.selectedInsights = JSON.stringify([]);
+                        row.dataset.unselectedInsights = JSON.stringify(data.result.insights);
+                    };
+                    
+                    insightsActions.appendChild(selectAllBtn);
+                    insightsActions.appendChild(clearAllBtn);
+                    
+                    insightsHeader.appendChild(insightsTitle);
+                    insightsHeader.appendChild(insightsActions);
+                    
+                    // Create insight tags
+                    const insightsList = document.createElement('div');
+                    insightsList.className = 'insights-list';
+                    
+                    // Clear the container and add the new elements
+                    insightsContainer.innerHTML = '';
+                    insightsContainer.appendChild(insightsHeader);
+                    insightsContainer.appendChild(insightsList);
+                    
+                    // Initialize selected insights
+                    const selectedInsights = [];
+                    const unselectedInsights = [];
+                    
+                    // Add insight tags
+                    data.result.insights.forEach(insight => {
+                        const tagElement = document.createElement('div');
+                        tagElement.className = 'insight-tag';
+                        tagElement.textContent = insight;
+                        tagElement.onclick = function() {
+                            tagElement.classList.toggle('selected');
+                            
+                            // Update selected insights
+                            updateSelectedInsights(row, insightsContainer);
+                        };
+                        
+                        // Add to selected by default
+                        tagElement.classList.add('selected');
+                        selectedInsights.push(insight);
+                        
+                        insightsList.appendChild(tagElement);
+                    });
+                    
+                    // Store initial selections
+                    row.dataset.selectedInsights = JSON.stringify(selectedInsights);
+                    row.dataset.unselectedInsights = JSON.stringify(unselectedInsights);
+                    
+                } else {
+                    insightsContainer.innerHTML = '<div class="error">No insights found</div>';
+                }
+            })
+            .catch(error => {
+                console.error('Error:', error);
+                insightsContainer.innerHTML = `<div class="error">Error extracting insights: ${error.message}</div>`;
+            });
+        }
+        
+        // Helper function to update selected insights stored in the row's dataset
+        function updateSelectedInsights(row, insightsContainer) {
+            const selectedInsights = [];
+            const unselectedInsights = [];
+            
+            const tags = insightsContainer.querySelectorAll('.insight-tag');
+            tags.forEach(tag => {
+                if (tag.classList.contains('selected')) {
+                    selectedInsights.push(tag.textContent);
+                } else {
+                    unselectedInsights.push(tag.textContent);
+                }
+            });
+            
+            row.dataset.selectedInsights = JSON.stringify(selectedInsights);
+            row.dataset.unselectedInsights = JSON.stringify(unselectedInsights);
+        }
+        
+        // Collect all selected insights from all rows
+        function collectAllSelectedInsights() {
+            const allSelectedInsights = [];
+            const queryItems = document.querySelectorAll('.query-item');
+            
+            queryItems.forEach(queryItem => {
+                const rows = queryItem.querySelectorAll('tbody tr:not(.insights-row)');
+                rows.forEach(row => {
+                    if (row.dataset.selectedInsights) {
+                        try {
+                            const selectedInsights = JSON.parse(row.dataset.selectedInsights);
+                            selectedInsights.forEach(insight => {
+                                if (!allSelectedInsights.includes(insight)) {
+                                    allSelectedInsights.push(insight);
+                                }
+                            });
+                        } catch (e) {
+                            console.error('Error parsing selected insights:', e);
+                        }
+                    }
+                });
+            });
+            
+            return allSelectedInsights;
+        }
+        
+        // Enhance the problem using the selected insights
+        function enhanceProblem() {
+            const problemInput = document.getElementById('userInput');
+            const originalProblem = problemInput.value.trim();
+            
+            if (!originalProblem) {
+                alert('Please provide a technical problem description first');
+                return;
+            }
+            
+            const selectedInsights = collectAllSelectedInsights();
+            if (selectedInsights.length === 0) {
+                alert('Please select at least one insight to enhance the problem');
+                return;
+            }
+            
+            // Get user comments if any
+            const userComments = document.getElementById('insightCommentArea')?.value || '';
+            
+            // If we have stored refined problems for this exact problem, just show them
+            const currentProblemKey = JSON.stringify({
+                problem: originalProblem,
+                insights: selectedInsights.sort(),
+                comments: userComments
+            });
+            
+            if (storedRefinedProblems[currentProblemKey]) {
+                showRefinedProblems(storedRefinedProblems[currentProblemKey]);
+                return;
+            }
+            
+            // Show loading overlay
+            const loadingOverlay = document.getElementById('globalLoadingOverlay');
+            if (loadingOverlay) {
+                loadingOverlay.style.display = 'flex';
+                loadingOverlay.querySelector('.progress-text').textContent = 'Enhancing problem statement...';
+            }
+            
+            // Send request to backend to refine the problem
+            fetch('/refine-problem', {
+                method: 'POST',
+                body: JSON.stringify({
+                    'original_problem': originalProblem,
+                    'insights': selectedInsights,
+                    'user_comments': userComments
+                }),
+                headers: { 'Content-Type': 'application/json' }
+            })
+            .then(response => response.json())
+            .then(data => {
+                // Hide loading overlay
+                if (loadingOverlay) loadingOverlay.style.display = 'none';
+                
+                if (data.error) {
+                    alert(`Error: ${data.error}`);
+                    return;
+                }
+                
+                if (data.result) {
+                    // Add current problem to history if not already there
+                    if (currentHistoryIndex === -1) {
+                        problemHistory.push(originalProblem);
+                        currentHistoryIndex = 0;
+                    }
+                    
+                    // Store the refined problems for future use
+                    storedRefinedProblems[currentProblemKey] = data.result;
+                    
+                    // Show refined problem options
+                    showRefinedProblems(data.result);
+                }
+            })
+            .catch(error => {
+                console.error('Error:', error);
+                if (loadingOverlay) loadingOverlay.style.display = 'none';
+                alert(`Error enhancing problem: ${error.message}`);
+            });
+        }
+        
+        // Display the refined problem options
+        function showRefinedProblems(result) {
+            const refinedContainer = document.getElementById('refinedProblemContainer');
+            const tabs = document.getElementById('refinedProblemTabs');
+            const content = document.getElementById('refinedProblemContent');
+            
+            // Clear previous content
+            tabs.innerHTML = '';
+            content.innerHTML = '';
+            
+            // Add tabs and content for each refined problem
+            let isFirst = true;
+            for (const key in result) {
+                if (result.hasOwnProperty(key)) {
+                    const problem = result[key];
+                    
+                    // Create tab
+                    const tab = document.createElement('div');
+                    tab.className = `refined-problem-tab ${isFirst ? 'active' : ''}`;
+                    tab.dataset.target = key;
+                    tab.textContent = `Option ${key.split('_')[1]}`;
+                    tab.onclick = function() {
+                        document.querySelectorAll('.refined-problem-tab').forEach(t => t.classList.remove('active'));
+                        document.querySelectorAll('.refined-problem').forEach(p => p.classList.remove('active'));
+                        tab.classList.add('active');
+                        document.getElementById(key).classList.add('active');
+                    };
+                    tabs.appendChild(tab);
+                    
+                    // Create content
+                    const problemDiv = document.createElement('div');
+                    problemDiv.className = `refined-problem ${isFirst ? 'active' : ''}`;
+                    problemDiv.id = key;
+                    
+                    const title = document.createElement('div');
+                    title.className = 'refined-problem-title';
+                    title.textContent = problem.title;
+                    
+                    const description = document.createElement('div');
+                    description.className = 'refined-problem-description';
+                    description.textContent = problem.description;
+                    
+                    const applyButton = document.createElement('button');
+                    applyButton.className = 'apply-problem-btn';
+                    applyButton.textContent = 'Apply This Problem';
+                    applyButton.onclick = function() {
+                        applyRefinedProblem(problem.description);
+                    };
+                    
+                    problemDiv.appendChild(title);
+                    problemDiv.appendChild(description);
+                    problemDiv.appendChild(applyButton);
+                    
+                    content.appendChild(problemDiv);
+                    
+                    isFirst = false;
+                }
+            }
+            
+            // Show the container
+            refinedContainer.style.display = 'block';
+            
+            // Scroll to the container
+            refinedContainer.scrollIntoView({ behavior: 'smooth' });
+        }
+        
+        // Apply the selected refined problem
+        function applyRefinedProblem(problemText) {
+            const problemInput = document.getElementById('userInput');
+            
+            // Add current problem to history
+            problemHistory.push(problemInput.value);
+            currentHistoryIndex = problemHistory.length - 1; // Set to latest version
+            
+            // Update problem text with the new version
+            problemInput.value = problemText;
+            
+            // Now add the new version to history
+            problemHistory.push(problemText);
+            currentHistoryIndex = problemHistory.length - 1; // Update index to point to the newly added version
+            
+            // Update history navigation display
+            updateHistoryNavigation();
+            
+            // Display is kept visible, removed: document.getElementById('refinedProblemContainer').style.display = 'none';
+            
+            // Focus on the problem input
+            problemInput.focus();
+            
+            console.log("Problem history after applying:", problemHistory, "Current index:", currentHistoryIndex);
+        }
+        
+        // Update the history navigation UI
+        function updateHistoryNavigation() {
+            const historyNav = document.getElementById('problemHistoryNav');
+            const prevBtn = historyNav.querySelector('.history-prev');
+            const nextBtn = historyNav.querySelector('.history-next');
+            const status = historyNav.querySelector('.history-status');
+            
+            // Update buttons state
+            prevBtn.classList.toggle('disabled', currentHistoryIndex <= 0);
+            nextBtn.classList.toggle('disabled', currentHistoryIndex >= problemHistory.length - 1);
+            
+            // Update status text
+            if (problemHistory.length > 0) {
+                status.textContent = `Version ${currentHistoryIndex + 1} of ${problemHistory.length}`;
+                historyNav.style.display = 'flex';
+            } else {
+                historyNav.style.display = 'none';
+            }
+        }
+        
+        // Navigate to previous problem version
+        function navigateProblemHistory(direction) {
+            const problemInput = document.getElementById('userInput');
+            console.log("Current history index:", currentHistoryIndex, "Total history:", problemHistory.length);
+            
+            if (direction === 'prev' && currentHistoryIndex > 0) {
+                currentHistoryIndex--;
+                problemInput.value = problemHistory[currentHistoryIndex];
+                console.log("Moving to previous version:", currentHistoryIndex);
+            } else if (direction === 'next' && currentHistoryIndex < problemHistory.length - 1) {
+                currentHistoryIndex++;
+                problemInput.value = problemHistory[currentHistoryIndex];
+                console.log("Moving to next version:", currentHistoryIndex);
+            }
+            
+            updateHistoryNavigation();
+        }
+        
+        function analyzePaper(paperUrl, queryItemElement) {
+            const patentBackground = document.getElementById('userInput').value.trim();
+            if (!patentBackground) {
+                alert('Please provide a patent background in the input field');
+                return;
+            }
+            
+            // Find the row containing this URL
+            const rows = queryItemElement.querySelectorAll('tbody tr');
+            let targetRow;
+            let documentType = 'pdf'; // Default type
+            
+            for (const row of rows) {
+                const urlLink = row.querySelector('.url-link');
+                if (urlLink && urlLink.href === paperUrl) {
+                    targetRow = row;
+                    // Get the document type from the first cell of the row
+                    documentType = row.cells[0].textContent.toLowerCase();
+                    break;
+                }
+            }
+            
+            if (!targetRow) {
+                alert('Could not find the paper in the results table');
+                return;
+            }
+            
+            // Check if we already have analysis columns
+            let scoreCell, justificationCell;
+            if (targetRow.cells.length <= 4) {
+                // We need to create the cells
+                scoreCell = document.createElement('td');
+                scoreCell.className = 'score-cell';
+                scoreCell.innerHTML = '<div class="loading-spinner"></div>';
+                
+                justificationCell = document.createElement('td');
+                justificationCell.className = 'justification-cell';
+                justificationCell.innerHTML = 'Analyzing...';
+                
+                targetRow.appendChild(scoreCell);
+                targetRow.appendChild(justificationCell);
+            } else {
+                // Use existing cells
+                scoreCell = targetRow.cells[4];
+                justificationCell = targetRow.cells[5];
+                scoreCell.innerHTML = '<div class="loading-spinner"></div>';
+                justificationCell.innerHTML = 'Analyzing...';
+            }
+            
+            // Send analysis request to backend
+            fetch('/analyze', {
+                method: 'POST',
+                body: JSON.stringify({ 
+                    'patent_background': patentBackground, 
+                    'pdf_url': paperUrl,
+                    'data_type': documentType
+                }),
+                headers: { 'Content-Type': 'application/json' }
+            })
+            .then(response => response.json())
+            .then(data => {
+                if (data.error) {
+                    scoreCell.innerHTML = 'Error';
+                    justificationCell.innerHTML = data.error;
+                } else if (data.result) {
+                    // Get score and justification from the result
+                    const result = data.result;
+                    const score = result.score !== undefined ? result.score : 'N/A';
+                    const justification = result.justification || 'No justification provided';
+                    
+                    // Update cells with results
+                    scoreCell.innerHTML = score;
+                    justificationCell.innerHTML = justification;
+                    
+                    // Color-code the score
+                    if (score >= 0 && score <= 5) {
+                        const redComponent = Math.floor((score / 5) * 255);
+                        const greenComponent = Math.floor(((5 - score) / 5) * 255);
+                        scoreCell.style.backgroundColor = `rgb(${redComponent}, ${greenComponent}, 0)`;
+                        scoreCell.style.color = 'white';
+                        scoreCell.style.fontWeight = 'bold';
+                        scoreCell.style.textAlign = 'center';
+                    }
+                    
+                    // Add "Extract Insights" button to the justify cell
+                    const insightsButton = document.createElement('button');
+                    insightsButton.textContent = 'Extract Insights';
+                    insightsButton.className = 'action-button insights-button';
+                    insightsButton.style.marginTop = '5px';
+                    insightsButton.style.fontSize = '0.8em';
+                    insightsButton.onclick = function(e) {
+                        e.preventDefault();
+                        extractInsights(paperUrl, targetRow);
+                    };
+                    justificationCell.appendChild(document.createElement('br'));
+                    justificationCell.appendChild(insightsButton);
+                } else {
+                    scoreCell.innerHTML = 'No data';
+                    justificationCell.innerHTML = 'Analysis failed';
+                }
+            })
+            .catch(error => {
+                console.error('Error:', error);
+                scoreCell.innerHTML = 'Error';
+                justificationCell.innerHTML = 'Failed to analyze paper';
+            });
+        }
+
+        // Update the placeholder text numbers after deletions
+        function updateQueryIndices() {
+            const queryItems = document.getElementById('queriesContainer').children;
+            for (let i = 0; i < queryItems.length; i++) {
+                const queryField = queryItems[i].querySelector('.query-field');
+                queryField.placeholder = `Search Query ${i + 1}`;
+            }
+        }
+        
+        // Get all queries as an array of strings
+        function getQueries() {
+            const queryFields = document.querySelectorAll('.query-field');
+            return Array.from(queryFields).map(field => field.value.trim());
+        }
+
+        // Analyze all unanalyzed papers in the results
+        function analyzeAllPapers() {
+            // Show loading overlay
+            const loadingOverlay = document.getElementById('globalLoadingOverlay');
+            if (loadingOverlay) loadingOverlay.style.display = 'flex';
+            
+            // Get all query items
+            const queryItems = document.querySelectorAll('.query-item');
+            let totalToAnalyze = 0;
+            let completedAnalyses = 0;
+            
+            // Count total papers that need analysis
+            queryItems.forEach(queryItem => {
+                const rows = queryItem.querySelectorAll('tbody tr');
+                rows.forEach(row => {
+                    if (row.cells.length <= 4 || row.cells[4].textContent.trim() === 'Error' || 
+                        row.cells[4].textContent.trim() === 'N/A' || row.cells[4].textContent.trim() === '') {
+                        totalToAnalyze++;
+                    }
+                });
+            });
+            
+            if (totalToAnalyze === 0) {
+                alert('No papers need analysis');
+                if (loadingOverlay) loadingOverlay.style.display = 'none';
+                return;
+            }
+            
+            // Function to update progress
+            function updateProgress() {
+                completedAnalyses++;
+                if (loadingOverlay) {
+                    const progressElement = loadingOverlay.querySelector('.progress-text');
+                    if (progressElement) {
+                        progressElement.textContent = `Analyzing papers: ${completedAnalyses}/${totalToAnalyze}`;
+                    }
+                }
+                
+                if (completedAnalyses >= totalToAnalyze) {
+                    if (loadingOverlay) loadingOverlay.style.display = 'none';
+                }
+            }
+            
+            // Analyze each paper that needs it
+            queryItems.forEach(queryItem => {
+                const rows = queryItem.querySelectorAll('tbody tr');
+                rows.forEach(row => {
+                    // Check if this row needs analysis
+                    if (row.cells.length <= 4 || row.cells[4].textContent.trim() === 'Error' || 
+                        row.cells[4].textContent.trim() === 'N/A' || row.cells[4].textContent.trim() === '') {
+                        
+                        // Get the URL
+                        const urlLink = row.querySelector('.url-link');
+                        const documentType = row.cells[0].textContent.toLowerCase();
+                        if (urlLink && urlLink.href) {
+                            // Create a promise for this analysis
+                            const promise = new Promise((resolve) => {
+                                // Prepare for analysis
+                                const patentBackground = document.getElementById('userInput').value.trim();
+                                
+                                // Create score and justification cells if needed
+                                let scoreCell, justificationCell;
+                                if (row.cells.length <= 4) {
+                                    scoreCell = document.createElement('td');
+                                    scoreCell.className = 'score-cell';
+                                    scoreCell.innerHTML = '<div class="loading-spinner"></div>';
+                                    
+                                    justificationCell = document.createElement('td');
+                                    justificationCell.className = 'justification-cell';
+                                    justificationCell.innerHTML = 'Analyzing...';
+                                    
+                                    row.appendChild(scoreCell);
+                                    row.appendChild(justificationCell);
+                                } else {
+                                    scoreCell = row.cells[4];
+                                    justificationCell = row.cells[5];
+                                    scoreCell.innerHTML = '<div class="loading-spinner"></div>';
+                                    justificationCell.innerHTML = 'Analyzing...';
+                                }
+                                
+                                // Send analysis request
+                                fetch('/analyze', {
+                                    method: 'POST',
+                                    body: JSON.stringify({ 
+                                        'patent_background': patentBackground, 
+                                        'pdf_url': urlLink.href,
+                                        'data_type': documentType
+                                    }),
+                                    headers: { 'Content-Type': 'application/json' }
+                                })
+                                .then(response => response.json())
+                                .then(data => {
+                                    if (data.error) {
+                                        scoreCell.innerHTML = 'Error';
+                                        justificationCell.innerHTML = data.error;
+                                    } else if (data.result) {
+                                        // Get score and justification
+                                        const result = data.result;
+                                        const score = result.score !== undefined ? result.score : 'N/A';
+                                        const justification = result.justification || 'No justification provided';
+                                        
+                                        // Update cells
+                                        scoreCell.innerHTML = score;
+                                        justificationCell.innerHTML = justification;
+                                        
+                                        // Color-code score
+                                        if (score >= 0 && score <= 5) {
+                                            const redComponent = Math.floor((score / 5) * 255);
+                                            const greenComponent = Math.floor(((5 - score) / 5) * 255);
+                                            scoreCell.style.backgroundColor = `rgb(${redComponent}, ${greenComponent}, 0)`;
+                                            scoreCell.style.color = 'white';
+                                            scoreCell.style.fontWeight = 'bold';
+                                            scoreCell.style.textAlign = 'center';
+                                        }
+                                    } else {
+                                        scoreCell.innerHTML = 'No data';
+                                        justificationCell.innerHTML = 'Analysis failed';
+                                    }
+                                    resolve();
+                                })
+                                .catch(error => {
+                                    console.error('Error:', error);
+                                    scoreCell.innerHTML = 'Error';
+                                    justificationCell.innerHTML = 'Failed to analyze paper';
+                                    resolve();
+                                });
+                            });
+                            
+                            // When this analysis is done, update the progress
+                            promise.then(() => {
+                                updateProgress();
+                            });
+                        } else {
+                            // No URL found, count as completed
+                            updateProgress();
+                        }
+                    }
+                });
+            });
+        }
+
+        // Remove rows with failed analyses
+        function removeFailedAnalyses() {
+            const queryItems = document.querySelectorAll('.query-item');
+            let removedCount = 0;
+            
+            queryItems.forEach(queryItem => {
+                const tbody = queryItem.querySelector('tbody');
+                const rows = Array.from(tbody.querySelectorAll('tr'));
+                
+                rows.forEach(row => {
+                    if (row.cells.length > 4) {
+                        const scoreText = row.cells[4].textContent.trim();
+                        if (scoreText === 'Error' || scoreText === 'N/A' || scoreText === 'No data') {
+                            tbody.removeChild(row);
+                            removedCount++;
+                        }
+                    }
+                });
+            });
+            
+            alert(`Removed ${removedCount} papers with failed analyses`);
+        }
+
+        // Export all tables to Excel
+        function exportToExcel() {
+            // Show loading overlay
+            const loadingOverlay = document.getElementById('globalLoadingOverlay');
+            if (loadingOverlay) {
+                loadingOverlay.style.display = 'flex';
+                loadingOverlay.querySelector('.progress-text').textContent = 'Generating Excel file...';
+            }
+            
+            try {
+                // Collect all data from all tables
+                const allData = [];
+                const queryItems = document.querySelectorAll('.query-item');
+                
+                queryItems.forEach(queryItem => {
+                    // Get the search query text for this table
+                    const queryText = queryItem.querySelector('.query-field').value;
+                    
+                    // Get all rows in this table
+                    const rows = queryItem.querySelectorAll('tbody tr:not(.insights-row)');
+                    
+                    rows.forEach(row => {
+                        // Skip empty rows or error rows
+                        if (row.cells.length === 1 && row.cells[0].colSpan > 1) {
+                            return; // Skip "No results found" rows
+                        }
+                        
+                        // Prepare object for this row
+                        const rowData = {
+                            'Topic': queryText,
+                            'Type': row.cells[0].textContent,
+                            'Title': row.cells[1].textContent,
+                            'Description': row.cells[2].textContent,
+                            'URL': row.querySelector('.url-link')?.href || ''
+                        };
+                        
+                        // Add score and justification if they exist
+                        if (row.cells.length > 4) {
+                            rowData['Score'] = row.cells[4].textContent;
+                            rowData['Justification'] = row.cells[5].textContent.split('\n')[0]; // Only get the justification text
+                        } else {
+                            rowData['Score'] = '';
+                            rowData['Justification'] = '';
+                        }
+                        
+                        // Add selected and unselected insights if they exist
+                        if (row.dataset.selectedInsights) {
+                            try {
+                                const selectedInsights = JSON.parse(row.dataset.selectedInsights);
+                                rowData['Selected Insights'] = selectedInsights.join('; ');
+                            } catch (e) {
+                                rowData['Selected Insights'] = '';
+                            }
+                        } else {
+                            rowData['Selected Insights'] = '';
+                        }
+                        
+                        if (row.dataset.unselectedInsights) {
+                            try {
+                                const unselectedInsights = JSON.parse(row.dataset.unselectedInsights);
+                                rowData['Unselected Insights'] = unselectedInsights.join('; ');
+                            } catch (e) {
+                                rowData['Unselected Insights'] = '';
+                            }
+                        } else {
+                            rowData['Unselected Insights'] = '';
+                        }
+                        
+                        allData.push(rowData);
+                    });
+                });
+                
+                // Check if we have any data
+                if (allData.length === 0) {
+                    if (loadingOverlay) loadingOverlay.style.display = 'none';
+                    alert('No data to export. Please perform a search first.');
+                    return;
+                }
+                
+                // Get all problem versions
+                const problemVersions = {};
+                if (problemHistory.length > 0) {
+                    problemHistory.forEach((problem, index) => {
+                        problemVersions[`Problem Version ${index + 1}`] = problem;
+                    });
+                } else {
+                    // If no history, just use the current problem
+                    const currentProblem = document.getElementById('userInput').value.trim();
+                    if (currentProblem) {
+                        problemVersions['Problem Version 1'] = currentProblem;
+                    }
+                }
+                
+                // Send to server for Excel generation
+                fetch('/export-excel', {
+                    method: 'POST',
+                    body: JSON.stringify({
+                        'tableData': allData,
+                        'userQuery': document.getElementById('userInput').value,
+                        'problemVersions': problemVersions
+                    }),
+                    headers: { 'Content-Type': 'application/json' }
+                })
+                .then(response => {
+                    if (!response.ok) {
+                        throw new Error(`Server error: ${response.status}`);
+                    }
+                    return response.blob();
+                })
+                .then(blob => {
+                    // Hide loading overlay
+                    if (loadingOverlay) loadingOverlay.style.display = 'none';
+                    
+                    // Create URL for the blob
+                    const url = window.URL.createObjectURL(new Blob([blob], {
+                        type: 'application/vnd.openxmlformats-officedocument.spreadsheetml.sheet'
+                    }));
+                    
+                    // Create a link and trigger download
+                    const a = document.createElement('a');
+                    a.href = url;
+                    a.download = 'patent_search_results.xlsx';
+                    document.body.appendChild(a);
+                    a.click();
+                    
+                    // Clean up
+                    window.URL.revokeObjectURL(url);
+                    document.body.removeChild(a);
+                })
+                .catch(error => {
+                    console.error('Error exporting to Excel:', error);
+                    if (loadingOverlay) loadingOverlay.style.display = 'none';
+                    alert(`Error exporting to Excel: ${error.message}`);
+                });
+            } catch (error) {
+                console.error('Error preparing Excel data:', error);
+                if (loadingOverlay) loadingOverlay.style.display = 'none';
+                alert(`Error preparing data for export: ${error.message}`);
+            }
+        }
+        
+        // Check if any insights are selected to enable/disable enhance button
+        function checkSelectedInsights() {
+            const selectedInsights = collectAllSelectedInsights();
+            const enhanceButton = document.getElementById('enhanceProblemButton');
+            
+            if (selectedInsights.length > 0) {
+                enhanceButton.classList.remove('disabled');
+                enhanceButton.disabled = false;
+            } else {
+                enhanceButton.classList.add('disabled');
+                enhanceButton.disabled = true;
+            }
+        }
+        
+        // Initialize comment area for insights
+        function initializeCommentArea() {
+            const container = document.getElementById('insightCommentContainer');
+            if (!container) return;
+            
+            const textarea = document.createElement('textarea');
+            textarea.id = 'insightCommentArea';
+            textarea.className = 'insight-comment-textarea';
+            textarea.placeholder = 'Add additional suggestions for enhancing the problem (optional)...';
+            
+            container.innerHTML = '';
+            container.appendChild(textarea);
+        }
+    </script>
+</head>
+<body>
+    <div class="container">
+        <header class="header">
+            <div class="logo">
+                <div class="logo-icon">P</div>
+                <h1>Patentability</h1>
+            </div>
+            <p>Enter a detailed description of your technical problem to generate search queries for finding relevant research papers.</p>
+        </header>
+
+        <section class="card">
+            <form id="queryForm" onsubmit="generateQueries(event)">
+                <div class="form-group">
+                    <label for="userInput">Technical Problem Description:</label>
+                    <div class="problem-history">
+                        <div id="problemHistoryNav" class="problem-history-nav" style="display: none;">
+                            <div class="history-arrow history-prev disabled" onclick="navigateProblemHistory('prev')">←</div>
+                            <div class="history-arrow history-next disabled" onclick="navigateProblemHistory('next')">→</div>
+                            <div class="history-status"></div>
+                        </div>
+                        <textarea id="userInput" placeholder="Describe your technical problem in detail..." required></textarea>
+                    </div>
+                </div>
+                
+                <button type="submit" class="btn btn-primary">Generate Search Queries</button>
+            </form>
+            
+            <div id="loadingIndicator">
+                <div class="loading-spinner"></div>
+                <p>Generating search queries... Please wait.</p>
+            </div>
+            
+            <div id="insightCommentContainer" class="insight-comment-area"></div>
+            
+            <div id="refinedProblemContainer" class="refined-problem-container" style="display: none;">
+                <div id="refinedProblemTabs" class="refined-problem-tabs"></div>
+                <div id="refinedProblemContent" class="refined-problem-content"></div>
+            </div>
+        </section>
+
+        <section class="search-options">
+            <label>Search Options:</label>
+            <div class="checkbox-group">
+                <div class="checkbox-item">
+                    <input type="checkbox" id="pdfOption" name="searchOptions" value="pdf" checked>
+                    <label for="pdfOption">PDF</label>
+                </div>
+                <div class="checkbox-item">
+                    <input type="checkbox" id="patentOption" name="searchOptions" value="patent">
+                    <label for="patentOption">Patent</label>
+                </div>
+                <div class="checkbox-item">
+                    <input type="checkbox" id="webOption" name="searchOptions" value="web">
+                    <label for="webOption">Web</label>
+                </div>
+            </div>
+        </section> 
+        
+        <section id="resultsContainer">
+            <h2>Generated Search Queries</h2>
+            
+            <div id="queriesContainer">
+                <!-- Query fields will be added here dynamically -->
+            </div>
+            
+            <div class="button-container">
+                <button type="button" class="btn btn-secondary" onclick="addQueryField()">
+                    <span>Add New Query</span>
+                </button>
+            </div>
+        </section>
+    </div>
+
+    <!-- Global action buttons (floating) -->
+    <div class="floating-buttons">
+        <button id="enhanceProblemButton" class="btn enhance-problem-button floating-button disabled" title="Enhance Problem using Selected Insights" onclick="enhanceProblem()" disabled>
+            Enhance Problem
+        </button>
+        <button id="analyzeAllButton" class="btn btn-primary floating-button" title="Analyze All Unanalyzed Papers">
+            Analyze All
+        </button>
+        <button id="removeFailedButton" class="btn btn-danger floating-button" title="Remove Papers with Failed Analyses">
+            Remove Failed
+        </button>
+        <button id="exportExcelButton" class="btn btn-success floating-button" title="Export All Data to Excel">
+            Export to Excel
+        </button>
+    </div>
+
+    <!-- Global loading overlay -->
+    <div id="globalLoadingOverlay" class="loading-overlay">
+        <div class="loading-content">
+            <div class="loading-spinner"></div>
+            <div class="progress-text">Processing...</div>
+        </div>
+    </div>
+
+    <script>
+        // Add event listeners for floating buttons
+        document.addEventListener('DOMContentLoaded', function() {
+            // Initialize the comment area
+            initializeCommentArea();
+            
+            // Check for selected insights periodically to enable/disable enhance button
+            setInterval(checkSelectedInsights, 1000);
+            
+            // Analyze all button
+            document.getElementById('analyzeAllButton').addEventListener('click', function() {
+                analyzeAllPapers();
+            });
+            
+            // Remove failed button
+            document.getElementById('removeFailedButton').addEventListener('click', function() {
+                removeFailedAnalyses();
+            });
+            
+            // Export to Excel button
+            document.getElementById('exportExcelButton').addEventListener('click', function() {
+                exportToExcel();
+            });
+        });
+    </script>
+</body>
+</html>