right java

app.py

@@ -6,37 +6,74 @@ import os
 if not os.path.exists('msmarco-passage'):
     os.system('python -c "from pyserini.search import LuceneSearcher; LuceneSearcher.from_prebuilt_index(\'msmarco-passage\')"')
 
-
-searcher = LuceneSearcher('msmarco-passage')
+searcher = LuceneSearcher.from_prebuilt_index('msmarco-passage')
 searcher.set_bm25(k1=0.9, b=0.4)
 
-
 def search_pyserini(query):
     try:
         hits = searcher.search(query, k=10)
         results = []
         for i, hit in enumerate(hits):
             doc = searcher.doc(hit.docid)
-            content = doc.raw()[:200] + "..." if len(doc.raw()) > 200 else doc.raw()
-            results.append(f"Rank: {i+1}\nDoc ID: {hit.docid}\nScore: {hit.score:.4f}\nContent: {content}\n")
-        
-        return "\n".join(results)
+            content = doc.raw()
+            results.append({
+                "rank": i + 1,
+                "doc_id": hit.docid,
+                "score": hit.score,
+                "content": content
+            })
+        return results
     except Exception as e:
-        return f"An error occurred: {str(e)}"
+        return [{"error": str(e)}]
+
+def format_results(results):
+    if isinstance(results, list) and len(results) > 0 and "error" in results[0]:
+        return f"<div class='error'>An error occurred: {results[0]['error']}</div>"
+    
+    html = "<div class='results-container'>"
+    for result in results:
+        html += f"""
+        <div class='result-item'>
+            <h3>Rank {result['rank']} (Score: {result['score']:.4f})</h3>
+            <p class='doc-id'>Doc ID: {result['doc_id']}</p>
+            <p class='content'>{result['content']}</p>
+        </div>
+        """
+    html += "</div>"
+    return html
 
 css = """
 .gradio-container {
     font-family: 'Arial', sans-serif;
 }
-.output-text {
-    white-space: pre-wrap;
-    font-family: 'Courier New', monospace;
-    font-size: 14px;
-    line-height: 1.5;
-    padding: 10px;
-    border: 1px solid #ccc;
-    border-radius: 5px;
-    background-color: #f9f9f9;
+.results-container {
+    display: flex;
+    flex-wrap: wrap;
+    gap: 20px;
+}
+.result-item {
+    border: 1px solid #ddd;
+    border-radius: 8px;
+    padding: 15px;
+    width: calc(50% - 10px);
+    box-shadow: 0 2px 4px rgba(0,0,0,0.1);
+}
+.result-item h3 {
+    margin-top: 0;
+    color: #333;
+}
+.doc-id {
+    font-size: 0.9em;
+    color: #666;
+    margin-bottom: 10px;
+}
+.content {
+    font-size: 0.95em;
+    line-height: 1.4;
+}
+.error {
+    color: red;
+    font-weight: bold;
 }
 """
 
@@ -55,16 +92,35 @@ with gr.Blocks(css=css) as iface:
         search_button = gr.Button("Search", variant="primary")
     
     with gr.Row():
-        output = gr.Textbox(
-            lines=20, 
-            label="Search Results", 
-            elem_classes=["output-text"]
-        )
+        output = gr.HTML(label="Search Results")
     
     search_button.click(
         fn=search_pyserini,
         inputs=query_input,
-        outputs=output
+        outputs=output,
+        _js="(results) => format_results(results)"  # Client-side formatting
     )
 
+# Add the JavaScript function to format results
+iface.load(js="""
+function format_results(results) {
+    if (Array.isArray(results) && results.length > 0 && results[0].hasOwnProperty('error')) {
+        return `<div class='error'>An error occurred: ${results[0].error}</div>`;
+    }
+    
+    let html = "<div class='results-container'>";
+    for (let result of results) {
+        html += `
+        <div class='result-item'>
+            <h3>Rank ${result.rank} (Score: ${result.score.toFixed(4)})</h3>
+            <p class='doc-id'>Doc ID: ${result.doc_id}</p>
+            <p class='content'>${result.content}</p>
+        </div>
+        `;
+    }
+    html += "</div>";
+    return html;
+}
+""")
+
 iface.launch()

packages.txt

@@ -1 +1 @@
-default-jre
+openjdk-21-jdk